决策树python sklearn 示例

原创 2017年10月11日 20:47:58

本文主要是使用python sklearn,完成决策树的demo,以及可视化,最终生成的决策树结果。

from sklearn.datasets import load_iris
from sklearn import tree
from sklearn.tree import export_graphviz
import subprocess


def visualize_tree(tree, feature_name, dot_file):
    """Create tree png using graphviz.
    tree -- scikit-learn DecsisionTree.
    feature_names -- list of feature names.
    dot_file -- dot file name and path
    """
    with open("tree.dot", 'w') as f:
        export_graphviz(tree, out_file=f,
                        feature_names=feature_name)

    dt_png = "dt.png"
    command = ["dot", "-Tpng", dot_file, "-o", dt_png]
    try:
        subprocess.check_call(command)
    except Exception as e:
        print e
        exit("Could not run dot, ie graphviz, to "
             "produce visualization")


def iris_demo():
    clf = tree.DecisionTreeClassifier()
    iris = load_iris()
    # iris.data属性150*4,iris.target 类别归一化为了0,1,2(150*1)
    clf = clf.fit(iris.data, iris.target)
    dot_file = 'tree.dot'
    tree.export_graphviz(clf, out_file=dot_file)
    visualize_tree(clf, iris.feature_names, dot_file)

    # (graph,) = pydot.graph_from_dot_file('tree.dot')
    # graph.write_png('somefile.png')


if __name__ == '__main__':
    iris_demo()
    pass

数据集


1. 花的分类的四种属性,150个示例

这里写图片描述

2. 花的分类,一共三类对应于0,1,2

这里写图片描述

3. 花的四个属性的描述

这里写图片描述

最终生成的结果:

这里写图片描述

pydot的安装见另一篇bolg

http://blog.csdn.net/haluoluo211/article/details/78200078

转载注明出处,并在下面留言!!!

参考

http://chrisstrelioff.ws/sandbox/2015/06/08/decision_trees_in_python_with_scikit_learn_and_pandas.html

http://www.kdnuggets.com/2017/05/simplifying-decision-tree-interpretation-decision-rules-python.html

python之sklearn

Scikit Learn: 在python中机器学习 Warning 警告:有些没能理解的句子,我以自己的理解意译。 翻译自:Scikit Learn:Machine Lea...

python之sklearn学习笔记

前言:本文是学习笔记。sklearn介绍scikit-learn是数据挖掘与分析的简单而有效的工具。 依赖于NumPy, SciPy和matplotlib。它主要包含以下几部分内容: 从功能来分: ...

python sklearn 分类算法简单调用

scikit-learn已经包含在Anaconda中。也可以在官方下载源码包进行安装。本文代码里封装了如下机器学习算法,我们修改数据加载函数,即可一键测试: [python] view p...
  • Bryan__
  • Bryan__
  • 2016年05月01日 00:58
  • 12701

python sklearn机器学习库 安装

1、准备工作 安装sklearn之前,我们需要先安装numpy,scipy函数库。 Numpy下载地址:http://sourceforge.net/projects/numpy/files/NumP...

Delphi7高级应用开发随书源码

  • 2003年04月30日 00:00
  • 676KB
  • 下载

Python-sklearn入门

Python-SKLearn入门本文主要来自Sklearn官网中的文档说明,详细文档请访问该网址[http://scikit-learn.org/dev/tutorial/basic/tutorial...

python sklearn包——3.1cross validation笔记

preface:做实验少不了交叉验证,平时常用from sklearn.cross_validation import train_test_split,用train_test_split()函数将数...

python 机器学习之sklearn-LR

from numpy import * from sklearn.datasets import load_iris # import datasets # load the datase...

Python机器学习库sklearn里利用决策树模型进行回归分析的原理

决策树的相关理论参考http://blog.csdn.net/cymy001/article/details/78027083 #原数据网址变了,新换的数据地址需要处理http://lib.stat....
  • cymy001
  • cymy001
  • 2017年11月17日 04:51
  • 31

Python sklearn库中决策树tree.DecisionTreeClassifier()函数参数介绍

Python sklearn库中决策树tree.DecisionTreeClassifier()函数参数介绍
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:决策树python sklearn 示例
举报原因:
原因补充:

(最多只允许输入30个字)