首页 > 其他 > 详细

sklearn中树模型可视化的方法

时间:2017-10-18 17:57:38      阅读:363      评论:0      收藏:0      [点我收藏+]

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

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中树模型可视化的方法

原文:http://www.cnblogs.com/ljy2013/p/7687965.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!