python决策树sklearn.tree、graphviz

古城微笑少年丶 2022-05-20 01:20 226阅读 0赞

![这里写图片描述][70]  
根据天气的好坏、是否是周末、是否促销来判断销量的高低，对此使用决策树来决策  
**数据预处理**

import pandas as pd
    filename = 'B:\pycharm\DataMining\data\sales_data.xls'
    data = pd.read_excel(filename, index_col='No')
    data[data == u'好'] = 1
    data[data == u'是'] = 1
    data[data == u'高'] = 1
    data[data != 1] = -1
    #print(data)
    x = pd.DataFrame(data.iloc[:, :3].astype(int))
    #x = data.iloc[:, :3].values.astype(int) 
    print(x)
    y = pd.DataFrame(data.iloc[:, 3].astype(int))
    print(y)

注意的是需要保留表头  
**建立决策树，采用的是基于信息熵的分析模型**

from sklearn.tree import DecisionTreeClassifier as DTC
    dtc = DTC(criterion='entropy') #建立决策树模型，基于信息熵
    dtc.fit(x, y)

**决策树可视化**

from sklearn.tree import export_graphviz #可视化决策树
    name = 'B:\pycharm\DataMining\data\\tree.dot'
    with open(name, 'w') as f:
        f = export_graphviz(dtc, feature_names=x.columns, out_file=f)

使用graphviz将dot文件转化为png文件  
![这里写图片描述][70 1]  
**打开png文件**  
![这里写图片描述][70 2]  
最上面的参与决策的标签(Weekend? Sale?)  
entropy是基于ID3算法的信息增益  
samples是这个决策下的样本总数目  
values是一个pairs：表示的是销量低的数据和销量高的数据的数目

[70]: /images/20220520/043996c0e84b44308ddd0e5bdd9cbd8c.png
[70 1]: /images/20220520/226ec2c623514dacb671e02d68ef1de8.png
[70 2]: /images/20220520/c8983f023ba44b9bbcb84dacb12d4464.png

发表评论取消回复

表情：

评论列表（有 0 条评论，226人围观）

还没有评论，来说两句吧...

相关阅读

相关 Python学习教程：决策树算法（三）sklearn决策树实战

面有跟大家出过两期关于决策树算法的**Python学习教程**，伙伴们学了学了，今天来点实际的吧，实践一把！做个巩固！ Python有一个著名的机器学习框架，叫skle...

桃扇骨/ 2024年04月17日 14:02/ 0 赞/ 51 阅读

相关决策树

1 决策树学习是以实例为基础的归纳学习算法，是应用最广泛的逻辑方法。 2 典型的决策树学习系统采用自顶向下的方法，在部分搜索空间中搜索解决方案。它可以确保求出一个简单的决策树

桃扇骨/ 2022年06月14日 04:27/ 0 赞/ 281 阅读

相关决策树

决策树是基于树结构来进行决策，这恰是人类在面临决策问题时一种很自然的处理机制。例如，我们要对“这是好瓜吗？”这样的问题进行决策时，通常会进行一系列的判断或“子决策”：我们先看“

旧城等待，/ 2022年05月25日 05:39/ 0 赞/ 342 阅读

相关决策树

一、决策树简介决策树是一种特殊的树形结构，一般由节点和有向边组成。其中，节点表示特征、属性或者一个类。而有向边包含有判断条件。如图所示，决策树从根节点开始延伸，经过不

骑猪看日落/ 2022年05月17日 00:55/ 0 赞/ 309 阅读

相关决策树

决策树：决策树是一个树形结构，每个非叶节点表示一个特征树形的测试，每个分支代表这个特征属性在某个值域上的输出，而叶节点存放一个类别。使用决策树进行决策的原理就是：从根

淩亂°似流年/ 2022年05月13日 08:50/ 0 赞/ 243 阅读

相关决策树

1 认识决策树如何高效的进行决策？特征的先后顺序（哪个特征先看，哪个特征后看） 2 决策树分类原理详解（看哪个特征能筛掉更多的数据，尽可能通过少

小咪咪/ 2022年04月23日 01:16/ 0 赞/ 237 阅读

相关决策树

决策树声明本文是来自网络文档和书本(周老师)的结合。概述决策树(Decision Tree）是在已知各种情况发生概率的[基础][Link 1]上，通

青旅半醒/ 2022年01月30日 06:49/ 0 赞/ 474 阅读

相关决策树

决策树对实例进行分类的树形结构，由节点和有向边组成。其实很像平时画的流程图。学习决策树之前要搞懂几个概念：熵：表示随机变量不确定性的度量，定义：H（p）=-![1409

冷不防/ 2021年09月30日 04:16/ 0 赞/ 505 阅读

相关决策树

熵的定义 ![5057999-5702853710d12e87.png][] 计算给定数据集的熵 def calcShannonEnt(dataSet):

客官°小女子只卖身不卖艺/ 2021年09月15日 06:34/ 0 赞/ 446 阅读

相关 python实现决策树

一、基本思想我们所熟知的决策树的形状可能如下：使用决策树算法的目的就是生成类似于上图的分类效果。所以算法的主要步骤就是如何去选择结点。划分数据集的最大...

灰太狼/ 2021年03月29日 13:43/ 0 赞/ 566 阅读