提取Rpart规则以分割数据集

时间:2017-09-24 07:20:25

标签: rpart

我使用Rpart来提出决策树,并使用预测,我能够应用规则来查看预测值。但我想按照生成的规则对数据集进行分段。基本上我想通过Rule / Rulenumber标记数据集中的行。怎么做到这一点?

1 个答案:

答案 0 :(得分:0)

问题发布已近一年,但可能对其他人有所帮助。 rpart树中的观察节点分配保存在tree$where

library("rpart")    
airq <- airquality[complete.cases(airquality),]
tree <- rpart(Ozone ~ ., data = airq)
tree$where