zoukankan      html  css  js  c++  java
  • sklearn中树模型可视化的方法

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

    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()) 
  • 相关阅读:
    SVN服务的配置与管理
    SVN配置多仓库与权限控制
    SVN使用详解
    这个问题他又来了,如何学编程!
    乘风破浪的程序员们
    Java 学习路线(史上最全 2020 版 ~ 持续更新中)
    P4782 【模板】2-SAT 问题
    HDU
    2020.8.3
    Interesting Computer Game
  • 原文地址:https://www.cnblogs.com/ljy2013/p/7687965.html
Copyright © 2011-2022 走看看