机器学习决策树(划分选择算法流程剪枝处理,连续值与缺失值处理)
Posted 夕述
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了机器学习决策树(划分选择算法流程剪枝处理,连续值与缺失值处理)相关的知识,希望对你有一定的参考价值。
决策树基于树结构来进行预测
决策树学习的目的是为了产生一棵泛化能力强,即处理未见示例能力强的决策树。
决策树学习的关键在于如何选择最优划分属性。一般而言,随着划分过程不断进行,我们希望决策树的分支结点所包含的样本尽可能属于同一类别,即结点的“纯度”(purity)越来越高。
划分选择
信息增益
信息熵
![](https://image.cha138.com/20230301/e309c36913d74af9aae6e0c3c23f7c5a.jpg)
信息熵越高,纯度越低
![](https://image.cha138.com/20230301/0746dab231604f9aba51839a078943b2.jpg)
信息增益
![](https://image.cha138.com/20230301/60ae63e574b74d5492770a2dcfd797e9.jpg)
一般而言,信息增益越大,则意味着使用属性 来进行划分所获得的“纯度提升”越大
信息增益对可取值数目较多的属性有所偏好
2. 增益率
![](https://image.cha138.com/20230301/41101917c1d544f1923ec523287ad125.jpg)
增益率准则对可取值数目较少的属性有所偏好
3. 基尼指数
![](https://image.cha138.com/20230301/42498228fd8f4022a360020322fd1ff9.jpg)
算法流程
一般的,决策树包含一个根结点、若干个内部结点和若干个叶结点。叶结点对应于决策结果,其他每个结点对应于一个属性测试;每个结点包含的样本集合根据属性测试的结果被分配到子结点中;根结点包含样本全集。从根结点到每个叶结点的路径对应了一个判定测试序列。
从根结点开始,对结点选择一个最优特征。由该特征的不同取值建立子结点,再对子结点递归地调用以上方法,直至(1)当前结点包含的样本全部属于同一类别(2)当前属性集为空,或所有样本在所有属性上取值相同,无法划分(3)当前结点包含的样本集合为空,不能划分。构建叶结点,最后每个子集都被分配到了叶结点上,即都有了明确的类,这样就生成了一棵决策树。
![](https://image.cha138.com/20230301/4e6b318678dd4bc4a1dbbd4649ddcdee.jpg)
剪枝处理
为什么剪枝
可通过“剪枝”来一定程度避免因决策分支过多,以致于把训练集自身的一些特点当做所有数据都具有的一般性质而导致的过拟合。
2. 预剪枝
决策树生成过程中,对每个结点在划分前先进行估计,若当前结点的划分不能带来决策树泛化性能提升,则停止划分并将当前结点记为叶结点,其类别标记为训练样例数最多的类别。
优点
降低过拟合风险
显著减少训练时间和测试时间开销
缺点
欠拟合风险:有些分支的当前划分虽然不能提升泛化性能,但在其基础上进行的后续划分却有可能导致性能显著提高。预剪枝基于“贪心”本质禁止这些分支展开,带来了欠拟合风险
3. 后剪枝
先从训练集生成一棵完整的决策树,然后自底向上地对非叶结点进行考察,若将该结点对应的子树替换为叶结点能带来决策树泛化性能提升,则将该子树替换为叶结点。
优点
后剪枝比预剪枝保留了更多的分支,欠拟合风险小,泛化性能往往优于预剪枝决策树
缺点
训练时间开销大:后剪枝过程是在生成完全决策树之后进行的,需要自底向上对所有非叶结点逐一考察
4. 例子
![](https://image.cha138.com/20230301/8892ebda96b14863bd1412e051312966.jpg)
预剪枝
![](https://image.cha138.com/20230301/83d5498b5ff3401f862b75848c6bc46a.jpg)
![](https://image.cha138.com/20230301/4cbb2ed84df34f4b8205c98c0f7fe2d8.jpg)
![](https://image.cha138.com/20230301/31d4e128c9bd4d93beb8cfc3ba381c23.jpg)
后剪枝
![](https://image.cha138.com/20230301/ca937ec679c542108675b2b17c6107f2.jpg)
![](https://image.cha138.com/20230301/71538b6f84d048bf80869a7fd023cc1f.jpg)
![](https://image.cha138.com/20230301/155605b667a34284bdae2af5eb2a892b.jpg)
连续值与缺失值
连续值处理
![](https://image.cha138.com/20230301/6acca838e7f4439d973e5a522a7b8e6b.jpg)
![](https://image.cha138.com/20230301/303d3fc6feb148a8bbf67205ee1f7af6.jpg)
与离散属性不同,若当前结点划分属性为连续属性,该属性还可作为其后代结点的划分属性
2. 缺失值处理
如何在属性缺失的情况下进行划分属性选择?
![](https://image.cha138.com/20230301/a1856d8c20314a5392a371e6b566c491.jpg)
例子:
![](https://image.cha138.com/20230301/71a982ac91dc4fe09b4155c512b2ab77.jpg)
![](https://image.cha138.com/20230301/11606839c2b5478988cd5b87a2f7c8b4.jpg)
给定划分属性,若样本在该属性上的值缺失,如何对样本进行划分?
![](https://image.cha138.com/20230301/96f1bbfb932a4bee9d94d13d32879ccd.jpg)
例子:
若划分属性选择为纹理:
![](https://image.cha138.com/20230301/41025e5298814b40993c3ef7d434c94a.jpg)
多变量决策树
![](https://image.cha138.com/20230301/30c33b0a1ae94cd9a012f1fc7b517960.jpg)
![](https://image.cha138.com/20230301/c1d8550488d5404e99daecc05dae174f.jpg)
以上是关于机器学习决策树(划分选择算法流程剪枝处理,连续值与缺失值处理)的主要内容,如果未能解决你的问题,请参考以下文章