正在阅读《第二节 决策树的ID3算法》
第(2/3)页


为了量化各种分裂属性选择方案的好坏,需要引入一个新的数学概念——熵(Entropy)。熵的概念是由信息论的创始人香农提出的,现在在很多学科中都有重要的用处。为了更容易理解这个概念,这里结合分类问题来描述它的定义。

设某个分类问题X包含n个类别m个特征属性,任意样本C属于第i个类别Xi(1≤i≤n)的概率为pi,则I(Xi)=-log2pi称为Xi的信息(Information),X的熵定义为

在分类问题中,如何简单地理解上述定义呢?显然有如下关系。

进而可以看出,类别Xi的信息是随着概率增加而减少的,也就是说,这个量衡量了Xi的不确定性。例如,概率为1时,它是注定会出现的,所以信息为0,不确定性最小;概率为0时,信息为+∞,不确定性最大。

而熵的概念可以结合概率中的全概率公式来理解,它代表了分类问题X的不确定性的期望(概率平均值),也就是对类别进行预测时的不确定性,或者说是预测的难度。

有了这样的认识,就可以很自然地给出确定分裂属性的原则:选择当前可以最大程度减少预测的不确定性的属性作为分裂属性。

衡量不确定性的减少程度有几种不同的计算方式,根据计算方式的不同,衍生出了不同的决策树算法。例如,使用信息增益的ID3算法,使用信息增益比的C4.5算法,还有使用基尼指数的CART算法。它们的原理是类似的,本教材介绍采用信息增益的ID3算法,其他算法只要把信息增益的计算公式替换成相应的其他计算公式就可以了。

在条件熵的基础上,就可以按如下方式定义信息增益(Gain)的概念了。

Gain(X,A)=H(X)-H(X|A)

信息增益还有一个名称叫作互信息,这是信息论中常用的名称。从熵和条件熵的定义可以看出,信息增益反映的是,选择A作为分裂属性后,分类问题的不确定性减少了多少。按照前面所说的选择分裂属性的原则,只要计算当前可供选择的所有特征属性对应的信息增益,以信息增益最大的属性作为决策树的下一个分裂属性即可。

对这个基本原则建立了清楚的认识后,使用信息增益就可以给出完整的构造决策树的ID3算法了。

第一步,初始化信息增益阈值。

第二步,生成根节点。通过计算分类问题的熵和不同特征属性对应的条件熵,选择信息增益最大并且大于增益阈值的属性作为第一个分裂属性生成根节点,并将该属性从候选属性中去除。

第三步,根据上层节点的不同取值,生成新的训练数据,计算不同的候选属性的信息增益,选择增益最大并且大于增益阈值的属性作为分裂属性生成新的节点,并将该属性从候选属性中去除。

第四步,重复第三步,直至满足终止条件。

该过程会重复使用第三步,这个方法被称为递归方法。现在还剩下一个问题需要说明,即与决策树构建有关的问题:何时停止树的生长?也就是在上述流程的最后一步中,终止条件是什么?常用的终止条件如下所列。

①如果所有属于分裂属性某个取值的训练数据都属于同一类别,则生成此类别的叶子节点并终止程序;

②如果特征属性已经使用完毕,即候选属性为空,则以当前训练数据中出现最多的类别生成叶子节点并终止程序;

③如果所有候选属性的信息增益都小于阈值,则以当前训练数据中出现最多的类别生成叶子节点并终止程序。

下面以上一节的电商消费数据为例,演示决策树ID3算法的实现过程,以帮助读者建立直观的认识。在实际操作中,一般以训练数据中相应的频率来代替概率。下面计算中四舍五入后都取两位小数,并且规定0×log20=1×log21=0。

(1)设定信息增益阈值

信息增益阈值设为0.03,认为小于这个阈值的增益对于改善分类的不确定性已经没有意义。

(2)生成根节点

H(X)=-{0.3×log2(0.3)+0.7×log2(0.7)}=0.88

相关小说

漂亮笨美人恋爱脑,甜追禁欲大佬 都市 / 尚未开放
漂亮笨美人恋爱脑,甜追禁欲大佬
今与紫
【追夫,非女强,甜甜的娇妻文学,不喜勿入!双洁,1v1。】 【身娇体软漂亮小美人X高冷禁欲闷骚霸总】 殷书漾对哥哥的好友谢观澜一见钟情后,就开启了追夫模式。 加好友,吃饭饭,去他家,求抱抱,..
25.08万字 14 天前
我死后第十三年 都市 / 尚未开放
我死后第十三年
不冻湖水
None是一支打破无数纪录的传奇乐队, 无数人以为这支乐队会一直长红,没想到终结在某次演唱会过后的雨夜—— 二十岁的主唱谢未雨从酒店窗台跌落,疑似自杀身亡。 从此四人变三人。 谢未雨出殡那..
43.2万字 14 天前
心之所屬 都市 / 尚未开放
心之所屬
C西溪C
心屬於你,便可輕易天荒地老。 最-新·书·刊:Ρo1❽sƒ。cοm(wᴏᴏ18.νⅰ&..
12.86万字 15 天前
阴嫁女,冥王夫 都市 / 尚未开放
阴嫁女,冥王夫
珺璟如烨
世界上有鬼吗?狗都不信!但是也不能把女儿卖给死人结冥婚吧!完了,世界上真有鬼让我给撞到了。还有更恐怖的吗?有!自从这以后,我还撞见了一大堆妖魔鬼怪!虽然眼前这个人,啊不,这个鬼,有钱有颜对我还特别好,但是,“鬼王大人,我真的不是你媳妇啊!”“小娘子,...
17.77万字 19 天前
好孕降临:我在兽世成了万人迷 都市 / 尚未开放
好孕降临:我在兽世成了万人迷
乳酪好吃
【雄竞+穿越+系统+爽文+兽世】末世特工木兮漫穿成远古兽世被退婚的“低生育力”白兔雌性,刚睁眼就听见雄狮未婚夫的嚣张羞辱要退婚。原主因此撞墙身亡。木兮漫冷笑打脸,夺回信物,怼翻绿茶。渣男气急败坏:你这生育力低下的废物,离了我,还有谁会要你!生育力底下?木...
28.44万字 19 天前
娇妻在上:墨少,轻轻亲 都市 / 尚未开放
娇妻在上:墨少,轻轻亲
十点听风
简介:池欢在婚礼的前一天晚上被自己的贴身保镖狠狠的睡了,一觉醒后,他单膝而跪,从容冷静,“我自首。” 她穿着男人的衬衫,慵懒而笑,“自首?”莹白的脚蹭着男人的腿,漫不经心的判决,“你只有两个选择,第一,自宫偿还我的清白,第二,做我的男人。”——池欢...
171.79万字 19 天前