python - 如何在 Databricks 上绘制 Python XGB 决策树

Question

我在数据 block 上绘制 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 作为一个包安装在数据 block 端。

Answer 1

我在数据 block 上显示 sklearn 决策树时遇到了类似的问题。

我的解决方案是更新 plot_tree 函数的源代码，以在通常返回的注释之外返回图形。

这是显示工作示例的数据 block 笔记本的摘录。我想 xgboost 也有类似的问题/设计。

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