前情代码:
from six import StringIO
import pydotplus
dot_data = StringIO()
tree.export_graphviz( # 导出sklearn的决策树的可视化数据
c45,
out_file=dot_data,
feature_names=feat_names,
class_names=['non-survival', 'survival'],
filled=True,
rounded=True,
impurity=False
)
# 用pydotplus生成图像
graph = pydotplus.graph_from_dot_data(
dot_data.getvalue().replace('\n', ''))
graph.write_png('tree.png')
报错:GraphViz’s executables not found
出错原因:没有安装GraphViz's executables
。Graphviz不是一个python tool,仍然需要安装GraphViz's executables
。
(1)下载链接 https://www.graphviz.org/download/
(2)解压后,编辑环境变量
- D:\BaiduNetdiskDownload\Graphviz-10.0.1-win64\bin
(3)测试是否安装成功
- win+R 输入
cmd
- 打开命令提示符
- 输入指令
dot -version
(4)重启jupyter,重新执行DecisionTree.ipynb