Визуализируйте XGB с помощью plot_tree на Macbook

Question

Я пытаюсь визуализировать дерево с помощью встроенной функции xgb «plot_tree» на Macbook:

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

I попробовал с: 1. brew install graphviz 2. установить перед в начале скрипта

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.