rpart 节点分配答案

【问题标题】：rpart node assignmentrpart 节点分配
【发布时间】：2012-04-05 13:09:48
【问题描述】：

是否可以提取适合的rpart 树的节点分配？当我将模型应用于新数据时会怎样？

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

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

【问题讨论】：

Search for corresponding node in a regression tree using rpart的可能重复
另请注意，对于用于拟合模型的数据（与新值的预测相反），您可以在summary.rpart 创建的对象的where 组件中找到此信息，或在拟合的 rpart 对象本身中。

标签： r decision-tree rpart

【解决方案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

【讨论】：

太好了，这正是我想要的。