在AWS Sagemaker上安装graphiz

Question

我在使用Python3的Jupyter笔记本上，并尝试使用以下代码绘制树：

import xgboost as xgb
from xgboost import plot_tree

plot_tree(model, num_trees=4)

在最后一行，我得到：

ModuleNotFoundError: No module named 'graphviz'
.........
.........
.........
~/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

您可以使用以下方法在笔记本实例上的环境中安装和卸载软件包：-

• 在jupyter实例上，转到“ Conda”选项卡。在那里你会发现 所有conda环境设置。
• 选择您的相关环境。选择选项后，在右面板下方-您将看到当前在所选环境中安装的所有软件包，在左面板中-您将找到所有可用的软件包并将其添加到您的环境中。< / li>
• 从左侧面板中，选择“ graphviz”，然后将所选软件包安装到当前环境中。
• 使用笔记本中“内核”下的“重新启动内核”选项重新启动内核。

或者，您可以在Jupyterhub笔记本中打开一个新终端，然后

1. 运行conda env list列出所有可用的环境/内核
2. 选择您要使用的版本（例如python3）并运行source activate python3
3. 现在运行所需的pip install ...命令
4. 一旦成功。运行source deactivate以关闭环境。然后打开带有内核的笔记本，然后尝试导入刚刚安装的模型。