機器學習之graphviz導出到png圖片

直入正題：

對於python進行決策樹編寫時，常常會遇到如何用graphviz如何生成png或者pdf文件：

這里是轉為png文件

具體操作如下：

1安裝庫

2生成dot文件

3將dot文件轉換為png文件

不需要多少代碼，也不需要多少時間。

一、安裝庫

首先需要導入幾個用到庫：

import graphviz

這里如果未能夠安裝用pip命令安裝即可：
windows命令下:
pip install graphvizot已經有許多文章介紹到了，這里就不再一一贅述，
如何安裝dot請參考：
https://jingyan.baidu.com/article/020278115032461bcc9ce598.html
下載參考：
下載地址在：http://www.graphviz.org/
當然如果系統仍然不能識別：

可以在代碼里面加入這一行：

os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'

注意把個人目錄看清

假若這里進行構建決策樹模型的名稱為Dtc

eg:

Dtc = DecisionTreeClassifier(max_depth=5,random_state=6)#構建決策樹模型
那么后面的操作就很容易了

dot_data = export_graphviz(Dtc,feature_names=['Pclass', 'Sex', 'Age'],class_names='Survived')
注意上述中的

feature_names、

class_names
名字要與后面的名字保持一致；

二、生成dot文件
關鍵一句來了：

with open("tree.dot", 'w') as f:
    f = export_graphviz(Dtc
                            , feature_names = ['Pclass', 'Sex', 'Age']  # tez
                            , class_names = 'Survived' # 類別名
                            , filled = True  # 由顏色標識不純度
                            , rounded = True  # 樹節點為圓角矩形
                            ,out_file=f)

上述一句將會在對應.py所在文件夾中產生一個tree.dot的文件，便於后面的生成png
三、將dot文件轉換為png文件
最后一步:生成png文件
在windows命令窗口中，先cd到對應的py文件夾中，然后輸入一下命令即完成由tree.dot文件轉化為png文件：
dot -Tpng tree.dot -o tree.png
這樣就得到了一個Png文件了，文件就在py所在的文件夾中。

文章部分轉自：

https://www.cnblogs.com/simpleDi/p/9971971.html