rpart节点分配

时间:2012-03-21 01:29:16

标签: r decision-tree rpart

是否可以为拟合的rpart树提取节点分配?当我将模型应用于新数据时呢?

我的想法是,我想使用节点作为聚类数据的方法。在其他包(例如SPSS)中,我可以保存预测的类,概率和节点号以供进一步分析。

鉴于R有多强大,我想有一个简单的解决方案。

1 个答案:

答案 0 :(得分:12)

尝试使用partykit包:

library(rpart)
z.auto <- rpart(Mileage ~ Weight, car.test.frame)
library(partykit)
z.auto2 <- as.party(z.auto)
predict(z.auto2, car.test.frame[1:3,], type = "node")

# Eagle Summit 4 Ford Escort   4  Ford Festiva 4 
#              7               7               7 
相关问题