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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章