rpubs - machine learning con r

¿Cómo hacer una parcela de árbol en el paquete caret?

Question

Estoy usando el paquete caret para modelar los datos usando el paquete rpart.

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

Como resultado, obtengo el objeto ''t'' y estoy intentando trazar este objeto para obtener el diagrama de árbol. Pero el resultado se ve así:

¿Hay alguna manera de hacer un diagrama de árbol a partir de un objeto de tren caret?

Answer 1

El objeto devuelto desde caret::train() es una lista. El elemento finalModel contiene su modelo.

Prueba esto:

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

Answer 2

Tuvo el mismo problema, pero las respuestas que se dieron aquí no lo resolverían, ya que usé un bosque al azar en lugar de un árbol, lo siguiente es que todos los que vienen aquí tienen el mismo problema:

En resumen: un árbol solo se puede mostrar cuando el método es algo como:

method = "rpart"

Usando un bosque al azar

method = "rf"

resultará en la siguiente trama:

Respuesta extendida ya aquí: Trazar el árbol de decisión en R (Caret)

Answer 3

parcela de árbol de aspecto más agradable:

library(rattle)
fancyRpartPlot(t$finalModel)