提取 Rpart 规则以分割数据集
Extracting Rpart rules to segment a dataset
我已经使用 Rpart 提出决策树，并使用预测，我能够应用规则来查看预测值。但是我想按照已经生成的规则对数据集进行分割。本质上我想用 Rule/Rulenumber 标记数据集中的行。如何做到这一点？
问题发布已近一年，但可能对其他人有所帮助。 rpart 树中的观察节点分配保存在 tree$where:
library("rpart")
airq <- airquality[complete.cases(airquality),]
tree <- rpart(Ozone ~ ., data = airq)
tree$where
我已经使用 Rpart 提出决策树，并使用预测，我能够应用规则来查看预测值。但是我想按照已经生成的规则对数据集进行分割。本质上我想用 Rule/Rulenumber 标记数据集中的行。如何做到这一点？
问题发布已近一年，但可能对其他人有所帮助。 rpart 树中的观察节点分配保存在 tree$where:
library("rpart")
airq <- airquality[complete.cases(airquality),]
tree <- rpart(Ozone ~ ., data = airq)
tree$where