How to make a tree plot in caret package? How to make a tree plot in caret package? r r

How to make a tree plot in caret package?


nicer looking treeplot:

library(rattle)fancyRpartPlot(t$finalModel)

enter image description here


The object returned from caret::train() is a list. The element finalModel contains your model.

Try this:

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

enter image description here


Had the same problem, but the answers given here wouldn't solve it, since I used a random forest instead of a tree, the following is for all coming here having the same issue:

In short: A tree can only be displayed when the method is something like:

method = "rpart"

Using a random forest

method = "rf"

will result in the following plot:enter image description here

Extended answer already here:Plot decision tree in R (Caret)