1.Window下安装GraphViz
1.1官网下载
https://graphviz.gitlab.io/_pages/Download/Download_windows.html
1.2.安装
下载msi然后安装,很简单,一路next, 记住安装路径(下图红色位置,注意格式),后面会设置环境变量
环境变量配置.png
在pycharm中运行pip install graphviz,重启pycharm即可。
2.决策可视化代码
from sklearn.datasets import load_iris
from sklearn import tree
import pydotplus
import graphviz
import os
# 导入路径
os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'
# 导入数据
iris = load_iris()
# 构建模型
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
# 保存模型
with open("d:\\user\\****\桌面\iris.dot", 'w') as f:
f = tree.export_graphviz(clf, out_file=f)
# 画图,保存到pdf文件
# 设置图像参数
dot_data = tree.export_graphviz(clf, out_file=None,
feature_names=iris.feature_names,
class_names=iris.target_names,
filled=True, rounded=True,
special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)
# 保存图像到pdf文件
graph.write_pdf("d:\\user\\***\桌面\iris.pdf")
网友评论