根据堆叠的插入符号模型绘制最终决策树

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)

enter image description here