fancyRpartPlot 决策树中节点顶部的数字是什么意思？

Question

fancyRpartPlot 决策树中节点顶部的数字是什么意思？我在下面的图片中突出显示了它们。

我的猜测是它们是节点的 order/rank，但我无法解释数字中的跳跃（在这个例子中，缺少 9-11）。

Answer 1

fancyRpartPlot 只是 prp 的包装。查看 prp 的源代码，这些是节点编号，由以下人员创建：

  if(nn || ni)
        draw.node.numbers(nn, ni, draw.shadows1, type, branch,
                Margin, xflip, yflip, cex,
                main, sub, col.main, cex.main, col.sub, cex.sub,
                xlim, ylim, node.xy, is.leaf, nodes,
                node.labs, font,  family, box.col, border.col, shadow.col,
                under.cex, under.font, under.ygap, ygap,
                split.labs, split.cex * cex, split.font, split.family, split.box.col,
                split.border.col, split.shadow.col,
                nn.cex, nn.font, nn.family, nn.col, nn.box.col,
                nn.border.col, nn.lty, nn.lwd, nn.round,
                split.adj, split.space, split.yspace, split.yshift,
                yshift, adj, space, yspace, shadow.offset,
                nn.adj, nn.yshift, nn.space, nn.yspace, bg)

list(node.boxes=node.boxes, split.boxes=split.boxes)

https://github.com/cran/rpart.plot/blob/master/R/prp.R

您还可以在代码中找到关于此的一些评论

Answer 2

树中每个节点顶部的数字对应于默认 print() 方法生成的树的文本表示中的分支编号。确认：

> dt <- rpart::rpart(Species ~ ., iris)
> print(dt)
n= 150 

node), split, n, loss, yval, (yprob)
      * denotes terminal node

1) root 150 100 setosa (0.33 0.33 0.33)  
  2) Petal.Length< 2.45 50   0 setosa (1.00 0.00 0.00) *
  3) Petal.Length>=2.45 100  50 versicolor (0.00 0.50 0.50)  
    6) Petal.Width< 1.75 54   5 versicolor (0.00 0.91 0.093) *
    7) Petal.Width>=1.75 46   1 virginica (0.00 0.022 0.98) *
> rattle::fancyRpartPlot(dt)

"jumps" 是 rpart() 调整树以移除一些分支的结果，而那些被修剪的分支不会出现在最终树中。

Answer 3

每个节点的顶部数字代表大多数category/class id。在您的情况下，起始节点 0.4（或 40%）是类别值“2”。

fancyRpartPlot 决策树中节点顶部的数字是什么意思？

What does the number on top of a node in a fancyRpartPlot decision tree mean?

r

rattle