文章目录
- 引入
- 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形式的：
发布者：admin，转转请注明出处：http://www.yc00.com/web/1734990791a3775925.html
评论列表（0条）