Метки не отображаются на графике дерева решений в - PullRequest
0 голосов
/ 29 ноября 2018

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

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

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

enter image description here

Ответы [ 2 ]

0 голосов
/ 30 ноября 2018

Используйте следующий код, как описано на странице справки для 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

0 голосов
/ 29 ноября 2018

Вам нужно

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