我认为在我的研究中解决这个问题时我非常接近。我正在为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)
答案 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
维护者询问此事。