决策树-可视化

1.Window下安装GraphViz

1.1官网下载

https://graphviz.gitlab.io/_pages/Download/Download_windows.html

graphviz.png

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")

3.效果

可视化决策树.png