小编avi*_*avi的帖子

从 rpart 包中的决策规则中提取信息

我需要从决策树中的规则中提取信息。我在 R 中使用 rpart 包。我在包中使用演示数据来解释我的要求:

data(stagec)
fit<- rpart(formula = pgstat ~ age + eet + g2 + grade + gleason + ploidy, data = stagec, method = "class", control=rpart.control(cp=0.05))
fit
Run Code Online (Sandbox Code Playgroud)

印花合身展示

n= 146 

node), split, n, loss, yval, (yprob)
      * denotes terminal node

 1) root 146 54 0 (0.6301370 0.3698630)  
   2) grade< 2.5 61  9 0 (0.8524590 0.1475410) *
   3) grade>=2.5 85 40 1 (0.4705882 0.5294118)  
     6) g2< 13.2 40 17 0 (0.5750000 0.4250000)  
      12) ploidy=diploid,tetraploid 31 11 0 (0.6451613 …
Run Code Online (Sandbox Code Playgroud)

r classification machine-learning decision-tree

2
推荐指数
1
解决办法
8904
查看次数