Метки не отображаются на графике дерева решений в

Question

Я использую пакет Caret для обучения моей модели.

Моя модель работает нормально.Но когда я строю дерево решений, метки остаются пустыми.Как я могу получить этикетки?

carMod <- train( FLAG ~.,data=df_train, method="rpart" )
plot(carMod$finalModel)

Answer 1

Используйте следующий код, как описано на странице справки для plot.rpart (https://stat.ethz.ch/R-manual/R-devel/library/rpart/html/plot.rpart.html):

plot(carMod$finalModel)
text(carMod$finalModel)

Альтернативно используйте

library(rpart.plot)
rpart.plot(carMod$finalModel)

Для получения дополнительной информации см. https://blog.revolutionanalytics.com/2013/06/plotting-classification-and-regression-trees-with-plotrpart.html

Answer 2

Вам нужно

plot(carMod$finalModel)
text(carMod$finalModel)