【发布时间】:2017-02-21 08:06:17
【问题描述】:
【问题讨论】:
标签: r decision-tree party
【问题讨论】:
标签: r decision-tree party
如果你想“看到”百分比,最简单的方法是使终端节点与响应的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
【讨论】:
vignette("partykit", package = "partykit") 和vignette("ctree", package = "partykit") 了解概述和一些指导。
partykit (1.2-0) 的 CRAN 版本,一切运行顺利。