Установите graphiz на AWS Sagemaker

Question

Я на ноутбуке Jupyter, использующем Python3, и пытаюсь построить дерево с кодом, подобным этому:

import xgboost as xgb
from xgboost import plot_tree

plot_tree(model, num_trees=4)

В последней строке я получаю:

~/anaconda3/envs/python3/lib/python3.6/site-packages/xgboost/plotting.py in to_graphviz(booster, fmap, num_trees, rankdir, yes_color, no_color, **kwargs)
196         from graphviz import Digraph
197     except ImportError:
--> 198         raise ImportError('You must install graphviz to plot tree')
199 
200     if not isinstance(booster, (Booster, XGBModel)):

ImportError: You must install graphviz to plot tree

Как мне установить graphviz, чтобы я мог видеть plot_tree?

Answer 1

Я наконец смог узнать, что у Конды есть пакет, который может установить его для вас. Я смог установить его, выполнив команду:

!conda install python-graphviz --yes

Обратите внимание, что --yes требуется только в том случае, если при установке требуется проверка добавления / изменения других пакетов, поскольку записная книжка Jupyter не является интерактивной, когда она работает.