r - 如何在插入符号包中制作树图？

Question

我正在使用 caret 包来使用 rpart 包对数据进行建模。

library('caret')
data(iris)
formula <- as.formula(Species ~.)
t <- train(formula,iris,method = "rpart",cp=0.002,maxdepth=8)
plot(t)

结果我得到了对象't'，我试图绘制这个对象来获得树状图。但结果看起来像这样：

有没有办法从插入符号火车对象制作树状图？

Answer 1

更好看的树形图：

library(rattle)
fancyRpartPlot(t$finalModel)

Answer 2

返回的对象caret::train()是一个列表。该元素finalModel包含您的模型。

试试这个：

plot(t$finalModel)
text(t$finalModel)

Answer 3

有同样的问题，但这里给出的答案并不能解决它，因为我使用的是随机森林而不是树，以下是所有来到这里有同样问题的人：

简而言之：只有当方法类似于：

method = "rpart"

使用随机森林

method = "rf"

将产生以下情节：

扩展答案已经在这里： 在 R 中绘制决策树（插入符号）