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