机器学习之graphviz导出到png图片


直入正题:

对于python进行决策树编写时,常常会遇到如何用graphviz如何生成png或者pdf文件:

这里是转为png文件

具体操作如下:

1安装库

2生成dot文件

3将dot文件转换为png文件

不需要多少代码,也不需要多少时间。

一、安装库

首先需要导入几个用到库:

import graphviz
这里如果未能够安装用pip命令安装即可:
windows命令下:
pip install graphvizot已经有许多文章介绍到了,这里就不再一一赘述,
如何安装dot请参考:
https://jingyan.baidu.com/article/020278115032461bcc9ce598.html
下载参考:
下载地址在:http://www.graphviz.org/
当然如果系统仍然不能识别:

可以在代码里面加入这一行:

os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'

注意把个人目录看清

假若这里进行构建决策树模型的名称为Dtc

eg:

Dtc = DecisionTreeClassifier(max_depth=5,random_state=6)#构建决策树模型
那么后面的操作就很容易了
dot_data = export_graphviz(Dtc,feature_names=['Pclass', 'Sex', 'Age'],class_names='Survived')
注意上述中的
feature_names、
class_names
名字要与后面的名字保持一致;
二、生成dot文件
关键一句来了:
with open("tree.dot", 'w') as f:
f = export_graphviz(Dtc
, feature_names = ['Pclass', 'Sex', 'Age'] # tez
, class_names = 'Survived' # 类别名
, filled = True # 由颜色标识不纯度
, rounded = True # 树节点为圆角矩形
,out_file=f)
上述一句将会在对应.py所在文件夹中产生一个tree.dot的文件,便于后面的生成png
三、将dot文件转换为png文件
最后一步:生成png文件
在windows命令窗口中,先cd到对应的py文件夹中,然后输入一下命令即完成由tree.dot文件转化为png文件:
dot -Tpng tree.dot -o tree.png
这样就得到了一个Png文件了,文件就在py所在的文件夹中。

文章部分转自:

https://www.cnblogs.com/simpleDi/p/9971971.html

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM