如何在插入符号包中绘制树图？

Question

我使用插入符号包对使用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

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

试试这个：

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

​

​

Answer 2

更好看的树状图：

library(rattle)
fancyRpartPlot(t$finalModel)

​

​

Answer 3

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

简而言之:只有当方法类似于以下内容时，才能显示树：

method = "rpart"

使用随机森林

method = "rf"

将产生以下图：

​

这里已经有了扩展答案：Plot decision tree in R (Caret)