在jupyter notebook中使用sklearn构建决策树时,作图用到graphviz
代码如下:
import graphviz
dot_data = tree.export_graphviz(clf, out_file=None)
graph = graphviz.Source(dot_data)
graph
报错信息如下:
no moudle named 'graphviz'
没有找到对应的moudle所以安装一下就成。
打开anaconda prompt, 输入命令:
conda install graphviz
如果这个没有成功,也可以使用pip进行安装
pip install graphviz
另外,如果电脑中已经安装了graphviz,就可以在jupyter中使用了。如果没的话,就去官网下载一个,windows选择stable版本。
安装完成之后,将%YourGraphvizPath%/bin
加入环境变量,重启jupyter notebook即可。