Окончательное дерево решений графика из модели сложенной каретки

Question

Я хотел бы построить дерево решений из окончательной модели сложенного каретного ансамбля.

Я управлял сложенным ансамблем с использованием каретки.Мой уровень 2 - rplot, в котором у меня есть финальная тестовая модель и часть предсказания.Но я не знаю, как построить фактический график дерева решений на основе этой модели.

моя окончательная модель теста:

test_model <- train(blenderData[,predictors], blenderData[,labelName],
                     method='rpart', trControl=myControl)

preds <- predict(object=test_model, testingData[,predictors])

Из этого я могу вычислить площадь под кривойно я хочу на самом деле построить дерево решений.Я не знаю, как это сделать.Кто-нибудь может помочь?Я очень новичок в этом.

Answer 1

Вы можете попробовать visNetwork пакет:

library(visNetwork)
library(caret)

test_model <- train(iris[,1:4], iris[,5],
                     method='rpart')

visTree( test_model$finalModel)

введите описание изображения здесь