我使用Rpart来提出决策树,并使用预测,我能够应用规则来查看预测值。但我想按照生成的规则对数据集进行分段。基本上我想通过Rule / Rulenumber标记数据集中的行。怎么做到这一点?
答案 0 :(得分:0)
问题发布已近一年,但可能对其他人有所帮助。 rpart树中的观察节点分配保存在tree$where
:
library("rpart")
airq <- airquality[complete.cases(airquality),]
tree <- rpart(Ozone ~ ., data = airq)
tree$where