rpart节点分配

Question

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

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

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

Answer 1

尝试使用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