如何在Databricks上绘制pythonxgbdecision树

提问者：小点点

我有一个问题，在数据库上绘制XgBooked决策树。在Python环境下工作时，会有一些奇怪的事情。

import xgboost as xgb
from xgboost import plot_importance

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

这会产生一个错误：

执行['point'，'-Tpng']失败，请确保Graphviz可执行文件位于您的系统路径上。

我已经作为一个包安装在数据库侧。

共1个答案

匿名用户

我遇到了一个类似的问题，即在数据库上显示skLearning决策树。

我的解决方案是更新plot_tree函数的源代码，以添加它通常返回的注释来返回数字。

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

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