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)

我试图在虹膜数据上完成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

有关更多示例，请参见rpart.plot小插图的第4章。