sklearn中樹模型可視化的方法

在機器學習的過程中，我們常常會用到樹模型的方式來解決我們的問題。在工業界，我們不僅要針對某個問題利用機器學習的方法來解決問題，而且還需要能力解釋其中的原理或原因。今天主要在這里記錄一下樹模型是怎么做可視化的方法：

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

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

2、window下需要安裝graphviz-2.38.msi。

安裝的時候采用默認安裝即可

3、執行如下代碼即可展示可視化

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())