我需要在决策树中提取每个叶子的路径和概率。这是一个快速的样本:
data(iris)
model<-rpart(Species~., data=iris)
summary(model)
我希望能够在摘要中提取所有这些信息,例如&#34;预测类&#34; ,&#34;班级数&#34;和&#34;概率&#34;以我可以放入表格的格式,而不是只能在控制台上读取它。
非常感谢!
答案 0 :(得分:1)
您需要访问&#34; model&#34;的属性。这是rpart.object的实例。 文档: https://stat.ethz.ch/R-manual/R-devel/library/rpart/html/rpart.object.html
例如插槽&#34;其中&#34; 有以下信息: 一个整数向量,其长度与根节点中的观察数相同,包含与每个观察所属的叶节点对应的帧的行号。
您可以在广告位&#34;框架&#34; 中找到的概率。