graphviz安裝（windows環境）

graphviz是一款開源的可視化圖形工具，在sklearn中，用於畫決策樹也很方便，其安裝流程如下：

下載

鏈接：https://graphviz.gitlab.io/_pages/Download/windows/graphviz-2.38.msi

安裝

雙擊下載好的安裝包，根據提示安裝，記住安裝目錄

添加環境變量

將一下目錄添加到環境變量中即可

# Graphviz的安裝目錄/bin
D:\Program Files (x86)\Graphviz2.38\bin

檢查是否安裝成功

打開命令行，輸入dot -version，如果出現如下圖所示的版本信息，說明安裝成功。

代碼測試

運行如下代碼，即可看到通過graphviz畫出的決策樹

from sklearn import tree
from sklearn.datasets import load_wine  # 紅酒數據
from sklearn.model_selection import train_test_split

wine = load_wine()
X_train, X_test, Y_train, Y_test = train_test_split(wine.data, wine.target, test_size=0.3)
clf = tree.DecisionTreeClassifier(criterion="entropy")
clf = clf.fit(X_train, Y_train)
score = clf.score(X_test, Y_test)  # 返回准確度

import graphviz
dot_data = tree.export_graphviz(clf
    ,feature_names = wine.feature_names  # 特征名
    ,class_names = wine.target_names  # 標簽名
    ,filled = True  # 顏色填充
    ,rounded = True  # 圓角邊框
)
graph = graphviz.Source(dot_data)
graph

參考鏈接：https://blog.csdn.net/lizzy05/article/details/88529483