2014-11-16 94 views
0

我需要提取决策树中每个叶子的路径和概率。这里有一个快速的样品一起工作:来自决策树的提取概率

data(iris) 

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

summary(model) 

enter image description here

我希望能够提取所有的汇总信息在这里,如“预测班”,“类罪名”和“概率“以我可以放入表格的格式,而不是只能在控制台上读取它。

非常感谢!

+1

'unclass(model)'有很多可能有用的东西。它看起来像是如果你挑选你想要的,表格可能不会太差 –

回答

0

您需要访问rpart.object的实例“model”的属性。 文档: https://stat.ethz.ch/R-manual/R-devel/library/rpart/html/rpart.object.html

例如时隙“其中”有以下信息: 相同的长度在根节点观测值的数量的整数向量,含有帧的对应于所述叶片的行数每个观察落入的节点。

你会发现在插槽“框架”的概率。

+0

这可能是正确的选择。虽然它不是非常直观,但我将不得不盯住它一会儿。他们在模型$ frame $ yval2中列出的概率与summary(model) – Jason

+0

中给出的概率不同,我在模型$ frame的以下列中看到了相同的概率:yval2.V5,yval2.V6,yval2.V7 –