如何在插入包中制作树形图？

Question

我使用插入包来使用rpart包对数据进行建模。

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

结果我得到了对象＆＃39;我试图绘制这个对象以获得树形图。但结果看起来像这样：

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

Answer 1

更好看的树形图：

library(rattle)
fancyRpartPlot(t$finalModel)

Answer 2

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

试试这个：

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

Answer 3

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

简而言之：只有在以下方法中才能显示树：

method = "rpart"

使用随机森林

method = "rf"

将导致以下情节：

此处已有扩展答案： Plot decision tree in R (Caret)