安装GraphViz
下载地址：https://graphviz.gitlab.io/_pages/Download/Download_windows.html
选择msi文件下载，安装即可。安装完成后，将bin目录加入到系统path环境变量中
安装graphviz的python库
使用conda install python-graphviz命令安装即可
生成可视化文件
import graphviz # doctest: +SKIP
from sklearn import tree
print(data.columns)
#ID3为决策树分类器fit之后得到的模型，注意这里必须在fit后执行，在predict之后运行会报错
dot_data = tree.export_graphviz(ID3, out_file=None,feature_names=data.columns[:-1],class_names=np.unique(y)) # doctest: +SKIP
graph = graphviz.Source(dot_data) # doctest: +SKIP
#在同级目录下生成tree.pdf文件
graph.render("tree") # doctest: +SKIP
参考http://sklearn.apachecn.org/cn/0.19.0/modules/tree.html#tree-algorithms
网友评论