从决策树中提取概率

时间:2014-11-16 21:41:04

标签: r decision-tree

我需要在决策树中提取每个叶子的路径和概率。这是一个快速的样本:

data(iris)

model<-rpart(Species~., data=iris)

summary(model)

enter image description here

我希望能够在摘要中提取所有这些信息,例如&#34;预测类&#34; ,&#34;班级数&#34;和&#34;概率&#34;以我可以放入表格的格式,而不是只能在控制台上读取它。

非常感谢!

1 个答案:

答案 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; 中找到的概率。