使用ctree在每个节点中显示音量，在R中绘制

Question

任何人都可以告诉我如何在每个节点中添加音量，而不是最终节点音量

t <- ctree(is_return ~ a + b + c)    
plot(t, type="simple")

我的树看起来像

如何修改该图，它将在每个圆节点上显示N =，而不仅仅是黑色或最终节点。

由于

Answer 1

这个想法是指定用于绘制内部节点的面板函数。

我生成一些数据和树

lls <- data.frame(N = gl(3, 50, labels = c("A", "B", "C")), 
                  a = rnorm(150) + rep(c(1, 0,150)),
                  b = runif(150))
pond= sample(1:5,150,replace=TRUE)
tt <- ctree(formula=N~a+b, data=lls,weights = pond)

自定义内部绘图功能。我画了一个圆圈，我写了一些重量。

innerWeights <- function(node){
  grid.circle(gp = gpar(fill = "White", col = 1))
  mainlab <- paste( node$psplit$variableName, "\n(n = ")
  mainlab <- paste(mainlab, sum(node$weights),")" , sep = "")
  grid.text(mainlab,gp = gpar(col='red'))
}

我绘制树

plot(tt, type='simple', inner_panel = innerWeights)

PS：结果取决于随机生成的数据，因此您可能无法得到相同的图。

Answer 2

导出决策树时，请更改图像的尺寸。对于具有200个终端节点的决策树，宽度应为约30.000。