R 函数获取 rpart 应用的规则

Question

iris <- read.csv("iris.csv") #iris data available in R
library(rpart)
iris.rpart <- rpart(Species~Sepal.length+Sepal.width+Petal.width+Petal.length, 
                 data=iris)
plotcp(iris.rpart)
printcp(iris.rpart)
iris.rpart1 <- prune(iris.rpart, cp=0.047)
plot(iris.rpart1,uniform=TRUE)
text(iris.rpart1, use.n=TRUE, cex=0.6)

我已经尝试在 iris 数据上完成 rpart。但是，是否可以通过使用 R 中的某些函数来获取 rpart 为当前树准备应用的规则，以便我们知道在向数据集添加更多新点时如何进行分类？

Answer 1

的 rpart.plot包有一个功能rpart.rules 用于为树生成一组规则。例如

library(rpart.plot)
iris.rpart <- rpart(Species~., data=iris)
rpart.rules(iris.rpart)

给予

   Species  seto vers virg
    setosa [1.00  .00  .00] when Petal.Length <  2.5
versicolor [ .00  .91  .09] when Petal.Length >= 2.5 & Petal.Width <  1.8
 virginica [ .00  .02  .98] when Petal.Length >= 2.5 & Petal.Width >= 1.8

和

options(width=1000)
rpart.predict(iris.rpart, newdata=iris[50:52,], rules=TRUE)

为您提供用于进行每个预测的规则:

   setosa versicolor virginica
50      1    0.00000  0.000000 because Petal.Length <  2.5
51      0    0.90741  0.092593 because Petal.Length >= 2.5 & Petal.Width <  1.8
52      0    0.90741  0.092593 because Petal.Length >= 2.5 & Petal.Width <  1.8

有关更多示例，请参阅第 4 章 rpart.plot vignette .