sklearn中樹模型可視化的方法

在機器學習的過程當中，咱們經常會用到樹模型的方式來解決咱們的問題。在工業界，咱們不只要針對某個問題利用機器學習的方法來解決問題，並且還須要能力解釋其中的原理或緣由。今天主要在這裏記錄一下樹模型是怎麼作可視化的方法：

一、首選須要用到幾個包，須要導入一下。沒有對應包的須要手動安裝一下。

from IPython.display import Image  
from sklearn import tree
import pydotplus 

二、window下須要安裝graphviz-2.38.msi。

安裝的時候採用默認安裝便可

三、執行以下代碼便可展現可視化

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

dot_data = tree.export_graphviz(elem, out_file=None,  #elem 是對應分類器
                         feature_names=select_featrue,   #對應的特徵
                         class_names=['0','1'],    #對應類別
                         filled=True, rounded=True,  
                         special_characters=True)  
graph = pydotplus.graph_from_dot_data(dot_data)  
graph.write_png('example.png')    #保存圖像
Image(graph.create_png())