2

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

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

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

4

1 回答 1

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
于 2012-03-21T02:41:25.153 回答