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