情节派对决策树

Question

我有如下图，你可以在图片中看到，有没有办法在叶节点中看到确切的百分比数？

Answer 1

如果你想“看到”百分比，最简单的方法是制作一个table()终端节点与响应，然后查看条件比例。

如果你想“看到”条形图中的比例，那么到目前为止还没有可能做到这一点。但是，我调整了node_barplot()函数以适应此功能。因此，如果您从R-Forge重新安装partykit包（party包的后继者），您可以尝试一下：

install.packages("partykit", repos = "http://R-Forge.R-project.org")
library("partykit")

为了说明，我将使用iris数据：

ct <- ctree(Species ~ ., data = iris)
plot(ct, tp_args = list(text = TRUE))

启用text = TRUE选项后，您可以在条形图上方（水平方向）绘制标签。等效规范为text = "horizontal"或text = "h"。如果你想要一个更窄的布局，你也可以使用：

plot(ct, tp_args = list(text = "vertical", ymax = 1.5))

频率表很简单：

tab <- table(predict(ct, type = "node"), iris$Species)
prop.table(tab, 1) * 100
##         setosa versicolor  virginica
##   2 100.000000   0.000000   0.000000
##   5   0.000000  97.826087   2.173913
##   6   0.000000  50.000000  50.000000
##   7   0.000000   2.173913  97.826087