sklearn中樹模型可視化的方法
阿新 • • 發佈:2017-10-18
方法 ron 問題 style 業界 們的 graphviz 還需要 plus
在機器學習的過程中,我們常常會用到樹模型的方式來解決我們的問題。在工業界,我們不僅要針對某個問題利用機器學習的方法來解決問題,而且還需要能力解釋其中的原理或原因。今天主要在這裏記錄一下樹模型是怎麽做可視化的方法:
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())
sklearn中樹模型可視化的方法