情节派对决策树

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