在 Macbook 上使用 plot_tree 可视化 XGB

Question

我正在尝试在 Macbook 上使用 xgb 的内置函数 "plot_tree" 可视化树：

import numpy as np
from xgboost import XGBRegressor, plot_tree

X = np.random.randint(0, 100, 100).reshape(100, 1)
y = np.random.randint(0, 500, 100).reshape(100, 1)

model = XGBRegressor(n_estimators = 1)

model.fit(X, y)


plot_tree(model)

但是，我收到以下错误：

ExecutableNotFound: failed to execute ['dot', '-Tpng'], make sure the Graphviz executables are on your systems' PATH

我试过： 1. brew 安装 graphviz 2.在脚本开头设置before

os.environ["PATH"] += os.pathsep + "/Users/alessandro/opt/anaconda3/bin/"

然而，似乎不起作用，报告同样的错误。

如何可视化我使用 Graphviz 或其他解决方案训练的单棵树？

Answer 1

您的 dot 命令似乎位于文件夹 /usr/local/bin/.

中

您可以使用以下终端命令将该文件夹全局添加到您的 PATH 中：

sudo nano /etc/paths
*type passsword*

然后将 /usr/local/bin 行添加到文件中，或者如果它已经存在，则将其移至列表顶部。

在此脚本中仅将该文件夹添加到您的 PATH 我认为您的命令非常接近，但需要稍作调整：

os.environ["PATH"] += os.pathsep + "/usr/local/bin/"

不是 dot 的遗漏，因为您正在尝试将文件夹而不是可执行文件添加到 PATH。

在 Macbook 上使用 plot_tree 可视化 XGB

Visualize XGB using plot_tree on Macbook

python

graphviz

xgboost