python 決策樹 視覺化

2021-10-11 01:51:52 字數 470 閱讀 1994

對於 cart 回歸樹的視覺化,可以先在電腦上安裝 graphviz;然後 pip install graphviz,這是安裝python的庫,需要依賴前面安裝的 graphviz。視覺化**如下:

from sklearn.tree import export_graphviz

import graphviz

# 引數是回歸樹模型名稱,不輸出檔案。

dot_data = export_graphviz(dtr, out_file=

none

)graph = graphviz.source(dot_data)

# render 方法會在同級目錄下生成 boston pdf檔案,內容就是回歸樹。

graph.render(

'boston')-

---

具體內容可以去 sklearn(

和 graphviz( 看看。

決策樹視覺化

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

python實現決策樹和視覺化決策樹

一.python實現決策樹 from sklearn import tree from sklearn.datasets import load wine from sklearn.model selection import train test split 匯入資料集,這裡用的是自帶的酒的資料 ...

Python決策樹demo視覺化

基礎環境 anaconda3 5.2.0 python3.6 win10x64 pip3 install graphviz pip3 install pydotplus coding utf 8 created on wed jul 31 16 51 08 2019 author 86182 fro...