可视化 C5.0 决策树?

机器算法验证 r 数据可视化 大车
2022-04-17 15:35:49

是否有直接的方法来可视化 c5.0 决策树?这是我的代码:

library(C50)
data(churn)
myTree = C5.0(x = churnTrain[, -20], y = churnTrain$churn)
summary(myTree)

我从一个我无法控制的程序中得到一长串 C5.0 对象,但如果绝对有必要,我可以将其转换为其他东西。任何想法/指针都受到高度赞赏。

1个回答

我可能在你的问题中遗漏了一些东西,但只是

plot(myTree)

为您提供树的可视化(基于 中的基础设施partykit

C5.0 树

当然这棵树很大,你要么需要放大图像,要么使用大屏幕阅读它......

您也可以使用partykit仅显示子树。例如,如果您只想显示根下方的左分支(从节点 2 开始)和根下方的右分支(从节点 33 开始),您可以这样做:

library("partykit")
myTree2 <- C50:::as.party.C5.0(myTree)
plot(myTree2[2])
plot(myTree2[33])