机器学习笔记（7）——C4.5决策树中的缺失值处理

最新推荐文章于 2024-04-28 22:37:13 发布

赵赵赵颖

最新推荐文章于 2024-04-28 22:37:13 发布

阅读量1.9w

点赞数 65

分类专栏：机器学习笔记文章标签： C4.5 机器学习决策树缺失值

本文链接：https://blog.csdn.net/leaf_zizi/article/details/83503167

版权

缺失值处理是C4.5决策树算法中的又一个重要部分，前面已经讨论过连续值和剪枝的处理方法：

现实任务中，通常会遇到大量不完整的样本，如果直接放弃不完整样本，对数据是极大的浪费，例如下面这个有缺失值的西瓜样本集，只有4个完整样本。

在构造决策树时，处理含有缺失值的样本的时候，需要解决两个问题：

（1）如何在属性值缺失的情况下选择最优划分属性？

（2）选定了划分属性，若样本在该属性上的值是缺失的，那么该如何对这个样本进行划分？

以上两个问题在周志华老师的《机器学习》书中有详细的讲解。但是还有一个问题：

（3）决策树构造完成后，如果测试样本的属性值不完整，该如何确定该样本的类别？

书中没有介绍，好在昆兰在1993年发表的文章中提供了解决方案。下面我们对以上3个问题逐一讨论。

之前的算法中，我们选择信息增益最大的属性作为最优划分属性，那么对于有缺失值的属性，其信息增益就是无缺失值样本所占的比例乘以无缺失值样本子集的信息增益。

$Gain(D,a)=\rho \times Gain(\tilde{D},a)$

其中 $\rho$ 是属性a上无缺失值样本所占的比例； $\tilde{D}$ 是属性a上无缺失值的样本子集。回顾一下ID3算法中，信息增益的计算方法：

$Ent(D)=-\sum_{k=1}^{|Y|}p_klog_2p_k$ $Gain(D,a)=Ent(D)-\sum_{v=1}^{V}\frac{|D^v|}{|D|}Ent(D^v)$

色泽：

$Ent(\tilde{D})=-(\frac{6}{14}log_2\frac{6}{14}+\frac{8}{14}log_2\frac{8}{14})=0.985$

另 $\tilde{D}^1,\tilde{D}^2,\tilde{D}^3$ 分别表示“色泽”属性上取值为“青绿”、“乌黑”、“浅白”的样本子集。

$Ent(\tilde{D^1})=-(\frac{2}{4}log_2\frac{2}{4}+\frac{2}{4}log_2\frac{2}{4})=1$

$Ent(\tilde{D^2})=-(\frac{4}{6}log_2\frac{4}{6}+\frac{2}{6}log_2\frac{2}{6})=0.918$

$Ent(\tilde{D^3})=-(\frac{0}{4}log_2\frac{0}{4}+\frac{4}{4}log_2\frac{4}{4})=0$

$Gain(\tilde{D},a) = 0.985-\left (\frac{4}{14}\times1+ \frac{6}{14}\times0.918+\frac{4}{14}\times0\right ) = 0.306$

$Gain(D,a)=\rho \times Gain(\tilde{D},a) = \frac{14}{17}\times0.306=0.252$

同样可以计算出其他几个属性的信息增益：

属性	色泽	根蒂	敲声	纹理	脐部	触感
信息增益	0.252	0.171	0.145	*0.424*	0.289	0.006

因此，选择“纹理”作为根节点进行划分。编号为{1,2,3,4,5,6,15}的7个样本进入“纹理=清晰”的分支，编号为{7,9,13,14,17}的5个样本进入“纹理=稍糊”的分支，编号为{11,12,16}的3个样本进入“纹理=模糊”的分支。

那么选定了划分属性，若样本在该属性上的值是缺失的，那么该如何对这个样本进行划分？

关注

专栏目录