首页 > 财经旅游 > 正文

小白也能看懂的机器学习中的决策树模型

2019-08-18 11:41:35   来源:东方头条   评论:0

决策树,顾名思义是一个数模型来的。从根节点开始一步步走到叶子节点 ,所有的数据最终都会落到叶子节点,既可以做分类也可以做回归 。下边用图表示一下。

在一个家族中,分布着年龄不一的人群。先按照年龄分类可分为两大类。然后再根据按照年龄分类的结果细分。

树的组成

树由根节点,非叶子节点与分支和叶子节点组成。根节点:第一个选择点 ,非叶子节点与分支:中间过程 ,叶子节点:最终的决策结果 。

决策树的训练与测试

训练阶段:从给定的训练集构造出来一棵树(从跟节点开始选择特征,

如何进行特征切分)

测试阶段:根据构造出来的树模型从上到下去走一遍就好了

一旦构造好了决策树,那么分类或者预测任务就很简单了,只需要走一遍就可以了,那么难点就在于如何构造出来一颗树,这就没那么容易了,需要考虑的问题还有很多的!

如何切分特征(选择节点)

问题:根节点的选择该用哪个特征呢?接下来呢?如何切分呢?

想象一下:我们的目标应该是根节点就像一个老大似的能更好的切分数据(分类的效果更好),根节点下面的节点自然就是二当家了。

目标:通过一种衡量标准,来计算通过不同特征进行分支选择后的分类情况,找出来最好的那个当成根节点,以此类推。

衡量标准-熵

熵:熵是表示随机变量不确定性的度量

(解释:说白了就是物体内部的混乱程度,比如杂货市场里面什么都有那肯定混乱呀,专卖店里面只卖一个牌子的那就稳定多啦)

公式:H(X)=- ∑ pi * logpi, i=1,2, ... , n

一个栗子:A集合[1,1,1,1,1,1,1,1,2,2]

B集合[1,2,3,4,5,6,7,8,9,1]

显然A集合的熵值要低,因为A里面只有两种类别,相对稳定一些而B中类别太多了,熵值就会大很多。(在分类任务中我们希望通过节点分支后数据类别的熵值大还是小呢?)

熵:不确定性越大,得到的熵值也就越大

当p=0或p=1时, H(p)=0,随机变量完全没有不确定性

当p=0.5时, H(p)=1,此时随机变量的不确定性最大

如何决策一个节点的选择呢?

信息增益:表示特征X使得类Y的不确定性减少的程度。(分类后的专一性,希望分类后的结果是同类在一起)

实例

数据:14天打球情况

特征:4种环境变化

目标:构造决策树

划分方式:4种

问题:谁当根节点呢?

依据:信息增益

在历史数据中( 14天)有9天打球, 5天不打球,所以此时的熵应为:

个特征逐一分析,先从outlook特征开始:

Outlook = sunny时,熵值为0.971

Outlook = overcast时,熵值为0

Outlook = rainy时,熵值为0.971

根据数据统计, outlook取值分别为sunny,overcast,rainy的概率分别为:5/14, 4/14, 5/14,熵值计算:5/14 * 0.971 + 4/14 * 0 + 5/14 * 0.971 = 0.693(分别计算一下其他三个类别的信息增益,这里不再一一计算了。gain(temperature)=0.029 gain(humidity)=0.152 gain(windy)=0.048)

信息增益:系统的熵值从原始的0.940下降到了0.693,增益为0.247

同样的方式可以计算出其他特征的信息增益,那么我们选择最大的那个

就可以啦,相当于是遍历了一遍特征,找出来了大当家,然后再其余的

中继续通过信息增益找二当家!

ID3: 信息增益(有什么问题呢?)

C4.5: 信息增益率(解决ID3问题,考虑自身熵)

CART:使用GINI系数来当做衡量标准

连续值怎么办?

决策树剪枝策略

为什么要剪枝:

决策树过拟合风险很大,理论上可以完全分得开数据(想象一下,如果树足够庞大,每个叶子节点不就一个数据了嘛)

剪枝策略可以分为预剪枝和后剪枝,预剪枝是边建立决策树边进行剪枝的操作,这种方法更实用一些。后剪枝是当建立完决策树后来进行剪枝操作。

预剪枝:限制深度,叶子节点个数,叶子节点样本数,信息增益量等

后剪枝:通过一定的衡量标准,(叶子节点越多,损失越大)

  责任编辑:

今日推荐

习近平在第五届中国国际进口博览会开幕式上发表致辞

11月4日晚,国家主席习近平以视频方式出席在上海举行的第五届中国国际进口博览会开幕式并发表题为《共创开放繁荣的美好未来》的致辞。新华社记者 李学仁 摄[详细]

中国共产党第二十次全国代表大会闭幕会在京举行

10月22日,中国共产党第二十次全国代表大会闭幕会在北京人民大会堂举行。新华社记者 陈建力 摄[详细]

天天学习|中南海月刊(2022.09)

天天学习金色九月,喜迎丰收。9月,一个个瞬间令人难忘:主持中央深改委会议决定健全关键核心技术攻关新型举国体制;出席中央军委晋升上将军衔仪式颁发命令状;新冠肺炎疫情发生以来首次出访...[详细]