R C5.0获得每个叶子的规则和概率

时间:2016-05-23 14:09:48

标签: r machine-learning decision-tree party

我认为在我的研究中解决这个问题时我非常接近。我正在为C5.0软件包寻找类似this的内容。

SO答案中提供的方法适用于party对象。但是,C5.0程序包不支持as.party。在我的进一步研究中,我发现this comment C5.0软件包的维护者已经编写了该函数,但没有将其导出。

我认为这应该有用,但遗憾的是建议的函数C50:::as.party.C5.0(mod1)会引发错误:

error in as.data.frame.default(x[[i]], optional = TRUE) : 
    cannot coerce class ""function"" to a data.frame

任何解决此错误的建议都表示赞赏。 让我们使用以下示例:

library(C50)
p = iris[1:4]
t = factor(iris$Species)
model = C50::C5.0(p,t)
#summary(model)

modParty = C50:::as.party.C5.0(model)

1 个答案:

答案 0 :(得分:3)

使用C5.0()的默认方法而不是公式方法时,似乎会出现问题。如果您使用后者,则as.party()转换成功运行,您可以应用所有方法:

model <- C5.0(Species ~ ., data = iris)
modParty <- C50:::as.party.C5.0(model)
modParty
## Model formula:
## Species ~ Sepal.Length + Sepal.Width + Petal.Length + Petal.Width
## 
## Fitted party:
## [1] root
## |   [2] Petal.Length <= 1.9: setosa (n = 50, err = 0.0%)
## |   [3] Petal.Length > 1.9
## |   |   [4] Petal.Width <= 1.7
## |   |   |   [5] Petal.Length <= 4.9: versicolor (n = 48, err = 2.1%)
## |   |   |   [6] Petal.Length > 4.9: virginica (n = 6, err = 33.3%)
## |   |   [7] Petal.Width > 1.7: virginica (n = 46, err = 2.2%)
## 
## Number of inner nodes:    3
## Number of terminal nodes: 4

然后在您链接的其他讨论中选择预测路径:

pathpred(modParty)[c(1, 51, 101), ]
##       response prob.setosa prob.versicolor prob.virginica
## 1       setosa  1.00000000      0.00000000     0.00000000
## 51  versicolor  0.00000000      0.97916667     0.02083333
## 101  virginica  0.00000000      0.02173913     0.97826087
##                                                              rule
## 1                                             Petal.Length <= 1.9
## 51  Petal.Length > 1.9 & Petal.Width <= 1.7 & Petal.Length <= 4.9
## 101                        Petal.Length > 1.9 & Petal.Width > 1.7

我不确定为什么该方法不适用于默认界面。但是,设置所需的模型框架可能更困难。不过,您可能会考虑向C50维护者询问此事。