如何在 Databricks 上绘制 Python XGBdecision 树答案

【问题标题】：How to Plot Python XGBdecision tree on Databricks如何在 Databricks 上绘制 Python XGBdecision 树
【发布时间】：2019-04-23 07:55:16
【问题描述】：

我在数据块上绘制 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 端。

【问题讨论】：

如前所述，您能否检查一下 Graphviz 是否在您的 PATH 环境变量中？即使安装了，也有可能调用不出来，因为你的系统不知道它在哪里。这对于 Keras 中的某些绘图功能也是必需的。

【解决方案1】：

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

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

这是一个数据块笔记本的摘录，展示了一个工作示例。我想 xgboost 有类似的问题/设计。

【讨论】：