ID3算法简介及基本原理
ID3算法基于信息嫡来选择最佳测试属性。它选择当前样本集中具有最大信息增益值的属性作为测试属性;样本集的划分则依据测试属性的取值进行,测试属性有多少不同取值就将样本集划分为多少子样本集,同时决策树上相应于该样本集的节点长出新的叶子节点。ID3算法根据信息论理论,采用划分后样本集的不确定性作为衡量划分好坏的标准,用信息增益值度量不确定性:信息增益值越大,不确定性越小。因此,ID3算法在每个非叶节点选择信息增益最大的属性作为测试属性,这样可以得到当前情况下最纯的拆分,从而得到较小的决策树。
设S是s个数据样本的集合。假定类别属性具有m个不同的值:C(i = 1,2…,m)。设s
是类C中的样本数。对一个给定的样本,它总的信息嫡为。
其中,Pi是任意样本属于Ci的概率,一般可以用。估计。
设一个属性A具有k个不同的值{a₁,a₂,···,ak},利用属性A将集合S划分为个子集{S₁,
S₂,···,Sa},其中S包含了集合S中属性A取a值的样本。若选择属性A为测试属性,则这些
子集就是从集合S的节点生长出来的新的叶节点。设Sij是子集Sj中类别为Ci的样本数,则
根据属性A划分样本的信息嫡值为
其中,
是子集Sj中类别为Ci的样本的概率。
最后,用属性A划分样本集S后所得的信息增益(Gain)为
Gain(A) = I(s₁,s₂,···,s_m) - E(A)
挺懵的,看例子先好了
餐饮案例
T餐饮企业作为大型连锁企业,生产的产品种类比较多,另外涉及的分店所处的位置也不同,数目比较多。对于企业的高层来讲,了解周末和非周末销量是否有大的区别,以及天气、促销活动这些因素是否能够影响门店的销量等信息至关重要。因此,为了让决策者准确了解和销量有关的一系列影响因素,需要构建模型来分析天气、是否周末和是否有促销活动对销量的影响,下面以单个门店为例来进行分析。
对于天气属性,数据源中存在多种不同的值,这里将那些属性值相近的值进行类别整合。如天气为“多云”,“多云转晴”,“晴”这些属性值相近,均是适宜外出的天气,不会对产品销量有太大的影响,因此将它们归为一类,天气属性值设置为“好”。同理,对于“雨”,“小到中雨”等天气,均是不适宜外出的天气,因此将它们归为一类,天气属性值设置为“坏”。
对于是否周末属性,周末则设置为“是”,非周末则设置为“否”。
对于是否有促销活动属性,有促销则设置为“是”,无促销则设置为“否”。
产品的销售数量为数值型,需要对属性进行离散化,将销售数据划分为“高”和“低”两类。将其平均值作为分界点,大于平均值的划分到类别“高”,小于平均值的划分为“低”类别。
采用ID3算法构建决策