整理自:https://mp.weixin.qq.com/s/LGyNq3fRlsRSatu1lpFnnw

    南大周志华老师的西瓜书—第四章


公式

信息增益是决策树ID3算法在进行特征切割时使用的划分准则,其物理意义和互信息完全相同,并且公式也是完全相同。

其中D表示数据集,A表示特征,信息增益表示得到A的信息而使得类X的不确定度下降的程度,在ID3中,需要选择一个A使得信息增益最大,这样可以使得分类系统进行快速决策。 

需要注意的是:在数值上,信息增益和互信息完全相同,但意义不一样,需要区分,当我们说互信息时候,两个随机变量的地位是相同的,可以认为是纯数学工具,不考虑物理意义,当我们说信息增益时候,是把一个变量看成是减少另一个变量不确定度的手段。

 

计算过程

 


 

以色泽属性为例,他有三个可能的取值{青绿,乌黑,浅白},对应{D1,D2,D3

  青绿 乌黑 浅白
正例 3 4 1
反例 3 2 4
总计 6 6 5

 

posted on 2018-05-17 22:30  笨拙的忍者  阅读(232)  评论(0编辑  收藏  举报