2014-11-16 157 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 –