r - 在R中建立分类树时，如何解释因子变量上的rpart拆分？

如果因子变量是“气候”，具有4个可能的值：“热带”，“干旱”，“温带”，“雪”，并且我的rpart树中的一个节点标记为“气候：ab”，那么分割是什么？

最佳答案

我假设您使用标准方式来绘制树

plot(f)
text(f)

正如您在text.rpart的帮助中所读到的那样，默认因子变量的参数pretty以字母表示，因此a表示levels(Climate)[1]，这意味着在左边的节点上可以观察到Climate==levels(Climate)[1]，在右边的节点上可以观察到。

您可以使用直接打印级别

plot(f)
text(f, pretty=1)

但我建议使用maptree package中的draw.tree：

require(maptree)
draw.tree(f)

我用假数据做图：

X <- data.frame(
    y=rep(1:4,25),
    Climate=rep(c("Tropical", "Arid", "Temperate", "Snow"),25)
)
f <- rpart(y~Climate, X)