如何在 Databricks 上绘制 Python XGBdecision 树

Question

我在数据块上绘制 Xgboost 决策树时遇到问题。XGboost 是为 python 安装的，这使得在这个环境中工作时有些奇怪。import xgboost as xgbfrom xgboost import plot_importancegbm=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