python决策树GraphViz可视化
安装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
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2018.04.19 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录