Как построить дерево Python XGBdecision для блоков данных

Question

У меня проблема с построением дерева решений Xgboost для блоков данных.XGboost установлен для Python, что делает его немного странным при работе в этой среде.

import xgboost as xgb
from xgboost import plot_importance

gbm=xgb.XGBClassifier().fit(X_train, y_train)
xgb.plot_tree(gbm)

Это выдает ошибку:

не удалось выполнить ['dot', '-Tpng'], убедитесь, что исполняемые файлы Graphviz находятся в вашей системе 'PATH.

У меня есть пакет graphviz, установленный на стороне блоков данных.

Answer 1

Я столкнулся с похожей проблемой с отображением дерева решений sklearn во время работы с блоками данных.

Мое решение состояло в том, чтобы обновить исходный код функции plot_tree, чтобы он возвращал фигуру в дополнение к аннотациям, которые она обычно возвращает.

Вот выдержка из блокнота данных, показывающая рабочий пример.Я думаю, что у xgboost есть похожая проблема / дизайн.

https://github.com/Foley-CJ/Tree_Plot/blob/master/Tree_Plot.ipynb