如何在数据库中绘制pythonxgbdecision树

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可执行文件位于系统的路径上。在

我把graphviz作为一个包安装在databricks端。在

Answer 1

我在使用databricks时遇到了一个类似的问题，即显示sklearn决策树。在

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

下面是一个databricks笔记本的摘录，展示了一个工作示例。我想xgboost也有类似的问题/设计。在

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