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

Question

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

Answer 1

我假设您使用标准方法来绘制树，即

plot(f)
text(f)

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

您可以直接使用打印级别

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

但我建议使用draw.treemaptree包：

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)