正在阅读《第二节 决策树的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

相关小说

宠妻狂魔:爹地,妈咪是我的 青春 / 完结
宠妻狂魔:爹地,妈咪是我的
一笔旧梦
惨遭妹妹陷害,她和陌生男人有了一夜情,五年后回归,却被腹黑又软萌的小娃娃被缠上。介绍一下,这是我爸,我觉得你们非常般配。总裁一脸冷酷将结婚证丢到了她面前,你已经是我的合法妻子了。什么情况?捡娃娃还带买一送一的?
27.59万字 4 年前
至尊魔妃,废柴大小姐 青春 / 完结
至尊魔妃,废柴大小姐
玖阑夜大人
她,是濮氏家族的濮四爷,打架找场子样样齐活。 她,潘家的废柴小姐,寄人篱下,受尽欺凌,样样占尽。 当她变成了她,功法仙器在手,宝贝丹药齐活。 敢算计她? 阴谋诡计,勾心斗角,全是她当年玩剩下的! 她定要让那些欺负她的人百倍偿还! 他,冷面仙君,修为超群,九重天门仙君之首,天帝都得给上几分薄面。 世人皆道他无心,偏偏他却认准了她,宠她入骨。 为渡她成仙,甘愿散去半生修为……
68.91万字 3 年前
顾先生,离婚吧! 青春 / 完结
顾先生,离婚吧!
冬菇家的猫
五年前,她被自己的丈夫陷害失身,家破人亡;为了报仇,她一人流落异国他乡,尝尽辛酸;四年后,她华丽回归,当名扬全城的美女设计师,碰上昔日里的前夫,又会擦出什么样的火花?
67.01万字 3 年前
嫡女重生,朕的皇后总想逃 都市 / 完结
嫡女重生,朕的皇后总想逃
涅凰
一朝眼瞎,一生凄苦,重活一世的沈姝誓要擦亮眼,防火防盗防渣男!渣男贱女,断她手脚筋,杀她长子,害她长女,毁她幼女!呵,这一桩桩,她都要同他们清算!渣男妄想攀附权贵?那就让他身败名裂!贱女阴毒?那就撕了她的美人皮!一心只有仇恨的她却无意间招惹到了一个了...
98.25万字 10 天前
农门冲喜小娘子 玄幻 / 完结
农门冲喜小娘子
笑猫嫣然
农门冲喜小娘子,相亲对像是初恋?呆了! 一口茶水没顺下去,嗝屁了。 再睁眼,床上躺着病歪歪的相公,她成冲喜娘子了? 屋子是借的,米缸是空的,地里长满荒草!!这到底是个什么情况?!! 床,只有一张?!被子,只有一条?! 白天伺候相公吃喝,晚上伺...
140.12万字 10 天前
星铁:重生成猫被符玄偷听心声 穿越 / 完结
星铁:重生成猫被符玄偷听心声
小太阳mt
++++琉光穿越到了的世界,但是本体却是一只流浪猫,并且还觉醒了一个好感度系统。该系统的作用是,让靠近琉光的人,都会对琉光产生好感。于是,第一个遇到琉光的符玄,一眼就喜欢上了这只圆滚滚、可爱的小猫,并将其带回了太卜司。但之后,符玄便发现,自己居然可以听见这小猫心里所想?!琉光:哎嘿嘿!符玄老婆今天又用她香香的、滑滑
158.16万字 7 天前