是否可以提取拟合的节点分配rpart
树?当我将模型应用于新数据时会怎样?
我的想法是,我想使用节点作为数据集群的一种方式。在其他软件包(例如 SPSS)中,我可以保存预测的类、概率和节点号以供进一步分析。
鉴于 R 的强大功能,我想有一个简单的解决方案。
尝试使用 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
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)