文章目录
- 引入
- 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形式的:
文章目录
- 引入
- 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形式的: