sklearn中树模型可视化的方法

Posted 2020-10-11 ljy2013

在机器学习的过程中，我们常常会用到树模型的方式来解决我们的问题。在工业界，我们不仅要针对某个问题利用机器学习的方法来解决问题，而且还需要能力解释其中的原理或原因。今天主要在这里记录一下树模型是怎么做可视化的方法：

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