決策樹之視覺化windows與linux兩個版本

2021-08-15 05:27:08 字數 1992 閱讀 1743

決策樹視覺化

決策樹的程式設計實戰

使用鳶尾花卉資料集訓練一棵決策樹,並繪製出來:

程式如下:

'''

使用鳶尾花卉資料集來訓練決策樹

'''from

sklearn.datasets

import

load_iris

from

sklearn

import

tree

iris=load_iris()

clf=tree.decisiontreeclassifier()

clf=clf.fit(iris.data,iris.target)

tree.export_graphviz(clf,

'/tree.dot'

)執行完上面程式後,會在

e:下生成乙個

tree檔案

如圖:

輸入win+r

,輸入框中現實的命令,如下所示:

執行完成後,會在

e:下生成乙個

pdf文件,如圖:

開啟pdf

便是繪製的決策樹,如圖:

注意,在這之前要先安裝好

graphviz

文件:解釋python**

安裝 graphviz

: 配置環境變數 轉化

dot檔案至

pdf視覺化決策樹:

dot.exe–tpdf e:/ tree.dot -o e:/tree.pdf

環境變數配置:

這樣就完成了整個過程。

如果是linux

作業系統,以

ubuntu

為例,首先安裝

graphviz

工具,輸入下面命令:

sudo apt-get install graphviz

便可完成安裝

然後,對於下面的程式執行完後:

from

sklearn.datasets

import

load_iris

from

sklearn

import

tree

iris=load_iris()

clf=tree.decisiontreeclassifier()

clf=clf.fit(iris.data,iris.target)

tree.export_graphviz(clf,

'./tree.dot')

會在當前目錄下生成乙個檔名為

tree.dot

檔案,如圖所示:

再在終端用

dot命令生成

pdf檔案即可,命令如下:

這樣便在目錄下產生了如上所示的tree.pdf

檔案,檔案內容與windows

中的一樣。

決策樹視覺化

決策樹相比其他演算法的乙個重要特性就是 可解釋性,構建決策樹的過程就相當於形成 if then 規則集。如果我們能夠將生成的決策樹視覺化,那麼我們就可以對資料集與 值之間的關係有清晰的認識。首先,我們以 sklearn.datasets 包中的鳶尾花資料集為例。from sklearn.datase...

sklearn 決策樹視覺化

來自google developer 的machine learning recipes with josh gordon youtube鏈結 這裡使用了lris flower data set 生成了乙個認花的小樹 import numpy as np from sklearn.datasets ...

Scikit learn 決策樹視覺化

平常網上看到的關於決策樹的scikit learn教程大都是直接給出容器的訓練 clf tree.decisiontreeclassifier clf clf.fit x,y clf.predict 然而有時候要的不只是 資料的結果,選擇決策樹還想要看到訓練出來的決策樹是什麼樣的。於是經過各種搜尋,...