admin 管理员组

文章数量: 887016

文章目录

      • 引入
    • Windows下载graphviz
    • 使用graphviz

引入

学习机器学习中决策树算法时,想要得到决策树结构,如:

而决策树中算法只能导出dot形式,要将dot形式转换成png图片形式,需要下载graphviz

Windows下载graphviz

网址:http://www.graphviz/download/#windows

点击下载,得到graphviz

双击打开安装




下载后文件Graphviz

且打开Windows命令行cmd,输入dot -version,得到以下graphviz版本内容:

即安装成功!!!

使用graphviz

from sklearn.tree import DecisionTreeClassifier,export_graphviz
# 用决策树进行预测
dec = DecisionTreeClassifier()
# 导出决策树的结构
export_graphviz(dec,out_file="./tree.dot",feature_names=['Age', 'Pclass', 'Sex=female', 'Sex=male'])

运行在当前文件夹下得到dot形式

在Windows命令行下,打开dot文件所在文件夹,输入命令语句:

dot -Tpng tree.dot -o tree.png
# 其中tree.dot为dot形式名称,tree.png为图片png形式名称

得到png图片形式

于是乎~~~~~~
dot形式的:

转换成了png形式的:

本文标签: 转换成 形式 图片 graphviz Windows