如何在 Databricks 上绘制 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 作为一个包安装在 databricks 端。

Answer 1

我在数据块上显示 sklearn 决策树时遇到了类似的问题。我的解决方案是更新 plot_tree 函数的源代码，以在通常返回的注释之外返回图形。这是一个数据块笔记本的摘录，显示了一个工作示例。我想 xgboost 有一个类似的问题/设计。https://github.com/Foley-CJ/Tree_Plot/blob/master/Tree_Plot.ipynb