码迷,mamicode.com
首页 > 其他好文 > 详细

sklearn中树模型可视化的方法

时间:2017-10-18 18:19:30      阅读:556      评论:0      收藏:0      [点我收藏+]

标签:window下   ipython   export   sel   pytho   +=   需要   splay   none   

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

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

标签:window下   ipython   export   sel   pytho   +=   需要   splay   none   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!