【机器学习】算法原理详细推导与实现(七):决策树算法

【机器学习】算法原理详细推导与实现(七):决策树算法

在之前的文章中,对于介绍的分类算法有逻辑回归算法朴素贝叶斯算法,这类算法都是二分类的分类器,但是往往只实际问题中y不仅仅只有\{0,1\},当出现一个新的类别y=2时,之前的分类器就不太适用,这里就要介绍一个叫做决策树的新算法,该算法对于多个目标的离散特征往往有比较好的分类效果,用以解决x是离散型的数据,这是判别模型,也是一个生成学习算法。

ID3决策树

假设当前网上有很多机器学习的教程,我想学机器学习,但是不知道哪一个教程比较好;又或者想买一个榴莲,不知道哪一只肉比较厚实,不知道该挑那一只,对于这种不确定性就叫做熵。

当一个事情有多种可能情况时,这件事情对于观察人而言具体是哪种情况的不确定性叫做熵。

信息

当你不知道选择哪一篇教程来学习机器学习时,有人和你说TTyb的机器学习教程很好,这个人对你提供的TTyb的机器学习教程很好就是信息;又或者在买榴莲的时候,卖榴莲的人和你说这个榴莲的肉肯定很厚实,卖榴莲的人和你说的这个榴莲的肉肯定很厚实就是信息。当某件事情存在不确定性,能够消除观察人对于某件事情的不确定性,也就是前面说的熵,能够消除不确定性的事物叫做信息。

信息有很多种表现形式,能够调整不确定的概率的事物可以叫做信息

例如卖榴莲的老板说有很大概率这个榴莲的肉可能很厚实;能够排除干扰的事物可以叫做信息,例如卖榴莲的老板说这个榴莲壳厚,肉肯定小不要选择这个;能够确定情况的事物可以叫做信息,卖榴莲的老板说这个榴莲肉肯定很厚实,开了不厚实算我的。

熵和信息数量相等,意义相反,获取信息意味着消除了不确定性熵。

信息熵

假设有4个榴莲只有一个肉是厚实的,老板和你说榴莲A的肉不厚实,给你提供了0.4bit的信息,榴莲B的肉不厚实,给你提供了0.6bit的信息,榴莲C的肉不厚实,给你提供了1bit的信息,最后让你确定了榴莲D肉是厚实的。老板每次都给你提供了一次信息,为什么提供的信息量却是不一样的?信息是如何量化的?回想一下什么东西有单位,质量、温度等物理量,信息也是一个物理量,要测量这个物理量,不妨回想一下我们是怎样测量质量的,“千克”最初又是怎么被定义出来的。

image

其实我们最初并不知道千克的质量,而是选定了一个参照物,把这个物体的质量就称为千克。当想要测量其他物体的质量时,就看这个物体的质量相当于多少个参照物体的质量,这里的“多少个”便是千克,如果参照物体换成“斤”,那么单位就会变化。

image

测量信息也是一样,既然信息消除的是不确定性,那么就选择另外一个事件的不确定性作为参照事件,当想要测量其他事件的信息时,就看待测事件的不确定性相当于“多少个”参照事件的不确定性,这里的多少个便是信息量。即待测物体质量为m,参照物体个数(信息量)为n,参照物体质量(千克)为x,那么:

m=n\times x

均匀分布的信息熵

当选择的参照物事件是像抛硬币这样,只有两种等概率情况的事件时,测得的信息量的单位就被称为比特(bit),然而测量参照物体个数(信息量)n时,我们是用待测物体质量m除以参照物体质量(千克)x,即:

n=\frac{m}{x}

可是测量信息时却不能用除法,因为抛掷3个硬币能够产生的等可能结果并非3\times 2=6,而是2^3=8种,也就是说信息量不是线性关系,而是指数关系:

|正正正|反正正|
|正反正|正正反|
|反反正|反正反|
|正反反|反反反|

所以当知道可能情况的个数m,想求这些情况相当于多少个n参照事件x所产生的,用指数运算的反函数,即对数运算计算:

n=log_xm

而这里硬币只有正面和反面,那么这里的参照事件x=2,公式变换为:

n=log_2m

上面的式子代表,在抛硬币的实验中,假设8个不确定情况就相当于3个硬币抛出来结果3=log_28,即信息量为3bit;4个不确定情况就相当于2个硬币抛出来结果2=log_24,即信息量为2bit

一般分布的信息熵

而上面讲的时抛掷硬币,被测事件的所有可能情况都是等概率事件,但是如果存在如下假设,这里有4个榴莲,只有一个榴莲的肉是厚实的,其他3个榴莲的肉是不厚实的

image

那么买到肉是厚实榴莲的不确定性(信息熵)为log_24=2bit,因为这里可能情况的个数m=4,所以不确定性是2bit。但是如果老板和你说最大的那个(最右边圆圈)榴莲有50%概率是肉厚的,那么从概率上来说买最大的那个(最右边圆圈)是肉厚的概率为\frac{1}{2},取出不是肉厚的概率是\frac{1}{3} \times \frac{1}{2},这里代表剩下3个榴莲要瓜分剩下的\frac{1}{2}概率,所以剩下3个榴莲每个的概率是 \frac{1}{\\6} ,这时候各个情况的概率不一样了:

image

这时该如何计算总信息量了呢?这时候就要分别测量每种可能情况的信息量后,乘以他们各自发生的概率再相加即可,即这时候买到肉厚的榴莲的不确定性(信息熵)为:

\frac{1}{6}log_2m_1+\frac{1}{6}log_2m_2+\frac{1}{6}log_2m_3+\frac{1}{2}log_2m_4

不过怎么测量每种情况的信息量log_2m呢,怎么知道概率为\frac{1}{6}的情况的不确定性相当于抛掷多少个硬币所产生的不确定性呢?我们确实没有办法用log_2m这个公式了,但是我们知道1%会发生的情况,相当于从100个等概率情况中确定实际情况,即p=1%=frac{1}{100},概率的倒数等于等概率情况的个数,即m=\frac{1}{100}=\frac{1}{p},用概率的倒数\frac{1}{p}替换等概率情况的个数m后,我们就可以计算每种情况的信息量了:

m_1=\frac{1}{\frac{1}{6}} ,m_2=\frac{1}{\frac{1}{6}},m_3=\frac{1}{\frac{1}{6}},m_4=\frac{1}{\frac{1}{2}}

再用每个情况的信息量乘以对应事件发生的概率,再相加后就能算出总信息量了:

\frac{1}{6}log_2\frac{1}{\frac{1}{6}}+\frac{1}{6}log_2\frac{1}{\frac{1}{6}}+\frac{1}{6}log_2\frac{1}{\frac{1}{6}}+\frac{1}{2}log_2\frac{1}{\frac{1}{2}}=1.79bit

由此可得到,任何随机变量X信息熵为:

H(X)=\sum_{i=1}^n p_ilog_2\frac{1}{p_i}

信息增益

既然任何随机变量X信息熵为:

H(X)=\sum_{i=1}^n p_ilog_2\frac{1}{p_i}

那么假设p是随机变量X发生的概率,如果存在多个变量XY,则他们的联合熵为:

H(X,Y)=\sum_{i=1}^n p(x_i,y_i)log_2p(x_i,y_i)

其中p(x_i,y_i)代表事件X=xY=y一起出现的联合概率。有了联合熵,又可以得到条件熵的表达式H(X|Y),条件熵类似于条件概率,当事件Y发生过后,事件X还剩下的不确定性(信息熵):

\begin{aligned} H(X|Y)&=\sum_{y} p(y)H(X|Y=y) \\ &=-\sum_{y} p(y)\sum_{x}p(x|y)log(p(x|y)) \\ &=-\sum_{y}\sum_{x}p(x,y)log(p(x|y)) \\ &=-\sum_{x,y}p(x,y)log(p(x|y)) \end{aligned}

其中H(X|Y)=H(X,Y)−H(Y),信息熵和信息增益的关系如下图所示:

image

左边的椭圆代表随机变量X的信息熵H(X),右边的椭圆代表随机变量Y的信息熵H(Y),中间重合的部分就是信息增益I(X,Y), 左边的椭圆去掉重合部分就是H(X|Y),右边的椭圆去掉重合部分就是H(Y|X),两个椭圆的并就是H(X,Y),由此可以得到信息增益的公式为:

I(X,Y)=H(X)−H(X|Y)

回到最初的问题,当4个榴莲中不知道那个是肉厚的榴莲的时候,随机变量X=肉厚,不确定性(信息熵)的套用公式为:

\begin{aligned} H(X)&=\sum_{i=1}^n p_ilog_2\frac{1}{p_i} \\ &=\frac{1}{4}log_2\frac{1}{\frac{1}{4}}+\frac{1}{4}log_2\frac{1}{\frac{1}{4}}+\frac{1}{4}log_2\frac{1}{\frac{1}{4}}+\frac{1}{4}log_2\frac{1}{\frac{1}{4}} \\ &=2 \end{aligned}

假设每个榴莲从左到右记为随机变量X1X2X3X4,老板告诉说最大的那个X4(最右边圆圈)榴莲有50%概率是肉厚的,即提供了信息Y,求肉厚的信息熵。这时候先计算的信息熵为条件熵H(X|Y)

\begin{aligned} H(X|Y)&=H(X1)+H(X2)+H(X3)+H(X4|Y=y) \\ &=\frac{1}{6}log_2\frac{1}{\frac{1}{6}}+\frac{1}{6}log_2\frac{1}{\frac{1}{6}}+\frac{1}{6}log_2\frac{1}{\frac{1}{6}}+\frac{1}{2}log_2\frac{1}{\frac{1}{2}} \\ &=1.79bit \end{aligned}

最后由总的不确定性(信息熵)减去后面的不确定性(信息熵),得到老板说的话“最大的那个(最右边圆圈)榴莲有50%概率是肉厚的”提供的信息为:

\begin{aligned} I(X,Y)&=H(X)−H(X|Y) \\ &=2-1.79 \\ &=0.21bit \end{aligned}

提供的信息0.21bit也叫做信息增益

ID3决策树算法步骤

输入的是m个样本,样本输出集合为X,每个样本有n个离散特征,特征集合即为Y,输出为决策树T。例如存在样本如下所示:

天气 气温 湿度 风力 外出
晴朗 高温 无风
晴朗 高温 有风
多云 高温 无风
下雨 温暖 无风
下雨 寒冷 正常 无风

则样本数量m=5;输出集合为或者,即X_1=是,X_2=否;每个样本的离散特征为n=4,即天气气温湿度风力;特征集合为Y,当离散特征Y'天气时,特征集合Y_i{晴朗,多云,下雨}。算法的过程为:

  1. 计算m中的各个特征(一共n个)对输出X的信息增益,选择信息增益最大的特征Y',其中Y' \in Y
  2. 按特征Y'的不同取值Y_i将对应的样本输出X分成不同的类别X_i,其中i \in n。每个类别产生一个子节点。对应特征值为Y_i。返回增加了节点的数T
  3. 对于所有的子节点,令X=X_i,m=m−\{Y'\}递归调用1-2步,得到子树T_j并返回。

C4.5决策树

ID3算法虽然提出了新思路,但是还是有如下4点需要改进的地方:

  1. ID3没有考虑连续特征,比如长度、密度都是连续值,无法在ID3运用,这大大限制了ID3的用途。--数据分箱即可解决
  2. ID3采用信息增益大的特征优先建立决策树的节点,但信息增益准则对可取值数目较多的属性有所偏好。
  3. 算法对于缺失值的情况没有做考虑。--特征工程即可解决
  4. 没有考虑过拟合的问题。--预剪枝和后剪枝

对于上面的第2点需要做一下解释。从上面可以知道,ID3信息增益的计算公式为:

I(X,Y)=H(X)−H(X|Y)

信息增益的大小取决于随机变量Y信息熵的大小,H(X|Y)越小则信息增益越大,而什么情况下H(X|Y)会有极小的信息熵呢?举一个极端的例子,如果上面的例子中,变量天气的特征互不相同变成:

天气 气温 湿度 风力 外出
大晴 高温 无风
小晴 高温 有风
多云 高温 无风
大雨 温暖 无风
小雨 寒冷 正常 无风

那么天气的信息熵会等于0,计算如下,当天气为大晴时,\frac{1}{1}的概率不外出,0的概率外出,那么大晴的信息熵H(X=外出|Y=大晴)为:

\begin{aligned} H(X=外出|Y=大晴)&=\frac{1}{1}log_2\frac{1}{\frac{1}{1}}+0log_20 \\ &=0 \end{aligned}

同理天气条件下其他特征的信息熵都是0,那么天气的信息熵H(X=外出|Y=天气)=0,得到天气的信息增益就最大了。也就是说决策树给天气这个属性下的离散特征都单独分成了一个子类,也就是说有多少种天气情况就有多少种取值,那么原数据集中有多少个离散特征,就会被划分为多少个子类。虽然这种划分毫无意义,但是从信息增益准则来讲,这就是最好的划分结果。

对于这种结果看似完美划分了随机变量的特征,然而根据大数定律,只有当样本数足够多的时候,频率才可以准确的近似概率。也就是说,样本数越少,对概率的估计结果的方差就会越大,结果也就越不准。想象一下做抛硬币实验来近似正面向上的概率,如果只抛两次,那么得到的正面向上的概率可能会非常离谱。而如果抛1万次,不论何时何地几乎总能得到近似0.5的概率。

而C4.5就是为了解决信息增益准则对可取值数目较多的属性有所偏好这种问题,由此提出了信息增益比,计算公式为:

Ir=(X,Y)=\frac{I(X,Y)}{IV(Y)}

其中IV(Y)计算方式为:

IV(Y)=\sum_yp_ylogp_y

p_y等于随机变量Y=y的概率。当信息熵H(X|Y)最小时,也就是分类各不相等的时候,IV(Y)也相对最小,这样就减轻了划分行为本身的影响。

cart决策树

基尼系数

有一个盒子里面放着小球,如果小球的颜色都是绿色,那么代表绿色的概率是p_绿=1,我们可以称随机变量X=绿是纯正的:

image

如果盒子里面的小球包含不同的颜色,不同颜色发生的概率为p_1,p_2,...,我们可以称随机变量X不纯正的:

image

假设某组样本存在多个随机变量X_1,X_2,...,X_n,他们各自发生的概率是\{p_1,p_2,...,p_n\},那么这组样本的纯正程度可以使用基尼系数(Gini)衡量:

Gini=1-p_1^2-p_2^2-...-p_n^2

假设一个盒子存在3种颜色的小球,他们的概率是:\{\frac{1}{3},\frac{1}{3},\frac{1}{3}\},那么基尼系数为:Gini=1-(\frac{1}{3})^2-(\frac{1}{3})^2-(\frac{1}{3})^2=0.6666。其他情况的基尼系数计算:

\{\frac{1}{10},\frac{2}{10},\frac{7}{10}\},\{1,0,0\}

计算的基尼系数为:

Gini=1-(\frac{1}{10})^2-(\frac{2}{10})^2-(\frac{7}{10})^2=0.46

Gini=1-(1)^2-(0)^2-(0)^2=0

基尼系数越大,代表样本越不纯净;基尼系数越小,代表样本越纯净,最终会找出Gini指数最小的来作为最优的划分点

剪枝

决策树算法为了避免过拟合和简化决策树模型,提出了剪枝的方法,剪枝分为预剪枝和后剪枝,剪枝的原理如下:

  • 预剪枝:在构造决策树的同时进行剪枝,也就是在节点划分前进行判断。所有决策树的构建方法,都是在无法进一步降低熵的情况下才会停止创建分支的过程,为了避免过拟合,可以设定一个阈值,熵减小的数量小于这个阈值,即使还可以继续降低熵,也停止继续创建分支。但是这种方法实际中的效果并不好。

  • 后剪枝:在决策树生长完全构造好了过后,对树进行剪枝。剪枝的过程是对拥有同样父节点的一组节点进行检查,判断如果将其合并,熵的增加量是否小于某一阈值。如果确实小,则这一组节点可以合并一个节点。后剪枝是目前最普遍的做法。

image

常见的剪枝有REP(Reduced Error Pruning)、PEP(Pessimistic Error Pruning)、CCP(Cost Complexity Pruning)和MEP(Minimum Error Pruning)算法,以下分别进行说明。

Reduced-Error Pruning(REP,错误率降低剪枝)

REP是最简单粗暴的一种后剪枝方法,其目的减少误差样本数量。该算法是从下往上依次遍历所有的子树,直至没有任何子树可以替换使得在验证集上的表现得以改进时,算法就可以终止。

假设树T存在很多结点t,其中t \in Te(t)表示节点t下训练样本误判的数量,则有训练样本误判总数量f(T)为:

f(T)=-\sum_{t \in T}e(t)

假设树T去掉某个节点t'变成树T',则有训练样本误判总数量f(T')为:

f(T')=-\sum_{t \in T'}e(t)

如果误判的数量降低,即剪枝之后使得误差降低:

f(T) \leqslant f(T')

那么代表剪枝后能使误差降低,剪枝成功。反复进行上面的操作,从底向上的处理结点,删除那些有害的结点,直到进一步修剪不能减低误差为止。例如:存在如下一棵树:

image

Step 1: 将节点4删掉替换成8和9,测试在验证集上的表现,若表现更好,则将节点4删掉并替换成8和9的并集,若表现不好则保留原树的形状
Step 2: 将节点2删掉替换成8、9和5,测试在验证集上的表现
Step 3: 将节点3删掉替换成6和7,测试在验证集上的表现

image

REP是最简单的后剪枝方法之一,不过由于使用独立的验证集,如果数据量较少(数据要分为训练集、验证集、测试集三份),那么在使用验证集进行剪枝的时候,可能会在验证集出现的稀有实例,却在测试集中没有出现,那么就会存在过度剪枝的情况。如果数据集较小,通常不考虑采用REP算法。尽管REP有这个缺点,但还是能够解决一定程度的过拟合问题。

Pesimistic-Error Pruning(PEP,悲观错误剪枝)

上文的REP方法思想简单且易于使用,不过最大的问题在于它需要一个新的验证集来修正我们的决策树在,PEP方法中不需要新的验证集,并且PEP是自上而下剪枝的。PEP的剪枝是把一颗子树(具有多个叶子节点)的分类用一个叶子节点来替代,如下所示:

image

原来子树1可以分成(4,5,6,7)多个类,但是剪枝变成只能分成一个类(1)。同样的样本子集,如果用子树分类可以分成多个类,准确率肯定要高一些,而用单颗叶子节点来分的话只能分成一个类,准确肯定要低一些,所以PEP的误判率肯定是上升的。训练数据也带来错分误差偏向于训练集,为了解决这一问题,给每个子叶节点加入修正项\frac{1}{2}。具有T个节点的树的错误率为:

E(T) = \sum_{t \in T}\frac{e(t)+1/2}{N(t)}

去掉节点K个子叶节点之后T′个节点的树的错误率为:

E(T')=\sum_{t \in T\, excep K}\frac{e(t)+1/2K}{N(t)}

其中e(t)表示节点t下训练样本误判的数量,N(t)表示节点t下训练样本的总数量。

PEP悲观错误剪枝剪枝法定义,如果 E(剪枝后误判数均值)−E(剪枝前误判数均值)<std(剪枝前误判数标准差),则可以剪枝,假设在子树中每一个样本的误判服从一个二项分布B(N,p),其中N表示子树所包含的所有样本个数。

所以,在剪枝前,其期望的误判数为:

e(t)=N(t)∗E(t)

其误判的标准差为:

std(t)=\sqrt{N(t)∗E(T)*(1-E(T))}

当剪枝后的误判数小于剪枝前的误判数一个标准差之后,就决定剪枝:

e(t')-e(t)<std(t)=\sqrt{N(t)\times E(T)\times (1-E(T))}

例如:存在如下一棵树:

image

对于节点T4而言,剪枝前后的计算过程如下:

已知E(T7)=0 ;E(T8)=2;E(T9)=1;E(T10)=1;E(T4)=7;N(t)=20,所以,剪枝前的误判概率为:

E(T)=\frac{(0+2+1+1)+4*0.5}{20}=0.3

所以:

e(t)=20*0.3=6

std(t)=\sqrt{20*0.3*(1-0.3)}=2.05

在我们对T4进行剪枝后,即将T4直接作为叶节点,剪枝后的误判概率:

E(T')=\frac{7+0.5}{20}=0.375

剪枝后的误判期望数为:

e(t')=20*0.375=7.5

7.5-2.05<6

因此满足条件,所以我们将把T4节点剪枝

Cost-Complexity Pruning(CCP,代价复杂度剪枝)

该算法为子树T_t定义了代价(cost)和复杂度(complexity),以及一个可由用户设置的衡量代价与复杂度之间关系的参数\alpha,其中,代价指在剪枝过程中因子树T_t被叶节点替代而增加的错分样本,复杂度表示剪枝后子树T_t减少的叶结点数,\alpha则表示剪枝后树的复杂度降低程度与代价间的关系,定义为:

α= \frac{R(t)−R(T_t)}{∣N∣−1}​

其中,

  • |N|:子树T_t中的叶节点数;
  • R(t):结点t的错误代价,计算公式为R(t)=r(t)*p(t);
  • r(t)为结点t的错分样本率,p(t)为落入结点t的样本占所有样本的比例;
  • R(T_t):子树T_t错误代价,计算公式为R(T_t)=\sum R(i)i为子树T_t的叶节点。

CCP算法可以分为两个步骤,
Step 1: 按照上述公式从下到上计算每一个非叶节点的αα值,然后每一次都剪掉具有最小αα值的子树。从而得到一个集合\{T_0,T_1,T_2,...,T_M\},其中,T_0表示完整的决策树,T_M表示根节点
Step 2: 根据真实的错误率在集合\{T_0,T_1,T_2,...,T_M\}选出一个最好的决策树

假设在下图中有数据100条,从下往上开始计算:

image

先以节点T4为例,那么可以得到:

R(t)=r(t)*p(t)=\frac{7}{9+7}*\frac{9+7}{100}=\frac{7}{100}

R(T_t)=\sum R(i)=R(7)+R(8)+R(6)=\frac{2}{3+2}*\frac{3+2}{100}+\frac{0}{0+2}*\frac{0+2}{100}+\frac{3}{6+3}*\frac{6+3}{100}=\frac{5}{100}

α= \frac{R(t)−R(T_t)}{∣N∣−1}​=\frac{\frac{7}{100}-\frac{5}{100}}{3-1}=\frac{1}{100}

REP PEP CCP
剪枝方式 自底向上 自顶向下 自底向上
计算复杂度 0(n) 0(n) o(n2)
误差估计 验证集上误差估计 使用连续纠正 标准误差

总结

对一般三种决策树进行总结,区分如下所示:

算法 支持模型 特征选择 连续值处理 确实值处理 剪枝
ID3 分类 信息增益 不支持 不支持 不支持
C4.5 分类 信息增益比 支持 支持 支持
CART 分类、回归 基尼系数 支持 支持 支持

实例

假设观测到历史有如下一组信息:

天气 气温 湿度 风力 外出
晴朗 高温 无风
晴朗 高温 有风
多云 高温 无风
下雨 温暖 无风
下雨 寒冷 正常 无风
下雨 寒冷 正常 有风
多云 寒冷 正常 有风
晴朗 温暖 无风
晴朗 寒冷 正常 无风
下雨 温暖 正常 无风
晴朗 温暖 正常 有风
多云 温暖 有风
多云 高温 正常 无风
下雨 温暖 有风

计算逻辑

可以看到这个树的根节点是外出,而根节点往下的一个节点,该怎么定义用哪一个条件呢?决策树构建树节点的条件在于:该节点能最大程度的提供有用的信息,使得不确定性(信息熵)减少得最快。也就是说,天气、气温、湿度、风力这四个条件,哪一个条件为结果外出提供更多的信息:

image

要计算哪一个条件能提供更多的信息,只要计算根节点和另一节点的信息熵差值,就是提供的信息增益了。由上面的表格可以知道,外出的概率为\frac{9}{14},不外出的概率是\frac{5}{14},很显然这是一个不均匀分布,要使用一般分布的信息熵计算:

\sum p_ilog_2\frac{1}{p_i}

则根节点外出的信息熵H(X=外出)为:

\begin{aligned} H(X=外出)&=\frac{9}{14}log_2\frac{1}{\frac{9}{14}}+\frac{5}{14}log_2\frac{1}{\frac{5}{14}} \\ &=0.940 \end{aligned}

接着分别计算各个条件的信息熵,先计算天气各个情况的信息熵:

  • 天气为晴朗时,2/5的概率外出,3/5的概率不外出,晴朗的信息熵H(X=外出|Y=晴朗)为:

\begin{aligned} H(X=外出|Y=晴朗)&=\frac{2}{5}log_2\frac{1}{\frac{2}{5}}+\frac{3}{5}log_2\frac{1}{\frac{3}{5}} \\ &=0.971 \end{aligned}

  • 天气为多云时,0的概率外出,0的概率不外出,信息熵H(X=外出|Y=多云)为:

\begin{aligned} H(X=外出|Y=多云)&=0+0 \\ &=0 \end{aligned}

  • 天气为下雨时,3/5的概率外出,2/5的概率不外出,信息熵H(X=外出|Y=下雨)为:

\begin{aligned} H(X=外出|Y=下雨)&=\frac{3}{5}log_2\frac{1}{\frac{3}{5}}+\frac{2}{5}log_2\frac{1}{\frac{2}{5}} \\ &=0.971 \end{aligned}

而天气是 晴朗 的概率为5/14,天气是 多云 的概率为4/14,天气是 下雨 的概率为5/14,所以 天气 的信息熵H(X=外出|Y=天气)为:

\begin{aligned} H(X=外出|Y=天气)&=\frac{5}{14} \times H(X=外出|Y=晴朗)+0 \times H(X=外出|Y=多云)+\frac{4}{14} \times H(X=外出|Y=下雨) \\ &=\frac{5}{14} \times 0.971+0 \times 0+\frac{4}{14} \times 0.971 \\ &=0.693 \end{aligned}

天气的信息增益I(X=外出,Y=天气)为:

\begin{aligned} I(X=外出,Y=天气)&=H(X=外出)-H(X=外出|Y=天气) \\ &=0.940-0.693 \\ &=0.247 \end{aligned}

同理计算得到,气温的信息熵H(X=外出|Y=气温)为:

\begin{aligned} H(X=外出|Y=气温)&=\frac{4}{14}H(X=外出|Y=高温)+\frac{6}{14}H(X=外出|Y=温暖)+\frac{4}{14}H(X=外出|Y=寒冷) \\ &=\frac{4}{14}(\frac{2}{4}log_2\frac{1}{\frac{2}{4}}+\frac{2}{4}log_2\frac{1}{\frac{2}{4}})+\frac{6}{14}(\frac{4}{6}log_2\frac{1}{\frac{4}{6}}+\frac{2}{6}log_2\frac{1}{\frac{2}{6}})+\frac{4}{14}(\frac{3}{4}log_2\frac{1}{\frac{3}{4}}+\frac{1}{4}log_2\frac{1}{\frac{1}{4}}) \\ &=\frac{4}{14}\times 1.0+\frac{6}{14}\times 0.918296+\frac{4}{14}\times 0.811278 \\ &=0.911 \end{aligned}

气温的信息增益I(X=外出,Y=气温)为:

\begin{aligned} I(X=外出,Y=气温)&=H(X=外出)-H(X=外出|Y=气温) \\ &=0.940-0.911 \\ &=0.029 \end{aligned}

湿度的信息熵H(X=外出|Y=湿度)为:

\begin{aligned} H(X=外出|Y=湿度)&=\frac{7}{14}H(X=外出|Y=高)+\frac{7}{14}H(X=外出|Y=正常) \\ &=\frac{7}{14}\times (\frac{6}{7}log_2\frac{1}{\frac{6}{7}}+\frac{1}{7}log_2\frac{1}{\frac{1}{7}})+\frac{7}{14}\times (\frac{3}{7}log_2\frac{1}{\frac{3}{7}}+\frac{4}{7}log_2\frac{1}{\frac{4}{7}}) \\ &=\frac{7}{14}\times 0.591673+\frac{7}{14}\times 1.159856 \\ &=0.788 \end{aligned}

湿度的信息增益I(X=外出,Y=湿度)为:

\begin{aligned} I(X=外出,Y=湿度)&=H(X=外出)-H(X=外出|Y=湿度) \\ &=0.940-0.788 \\ &=0.152 \end{aligned}

风力的信息熵H(X=外出|Y=风力)为:

\begin{aligned} H(X=外出|Y=风力)&=\frac{6}{14}E_{有风}+\frac{8}{14}E_{无风} \\ &=\frac{6}{14}\times (\frac{3}{6}log_2\frac{1}{\frac{3}{6}}+\frac{3}{6}log_2\frac{1}{\frac{3}{6}})+\frac{8}\\{14}\times (\frac{6}{8}log_2\frac{1}{\frac{6}{8}}+\frac{2}{8}log_2\frac{1}{\frac{2}{8}}) \\ &=\frac{6}{14}\times 1.0+\frac{8}{14}\times 0.811278 \\ &=0.892 \end{aligned}

风力的信息增益I(X=外出,Y=风力)为:

\begin{aligned} I(X=外出,Y=风力)&=H(X=外出)-H(X=外出|Y=风力) \\ &=0.940-0.892 \\ &=0.048 \end{aligned}

可以看到,信息增益:I(X=外出,Y=天气)>I(X=外出,Y=湿度)>I(X=外出,Y=风力)>I(X=外出,Y=气温),所以对是否外出提供信息最多的条件是天气。也就是说在决定是否外出的时候,天气是最重要的一个影响因素,天气的好坏很大程度决定了当天是否会外出,减少是否外出的不确定性。

但是天气分为多云、下雨、晴朗,由数据集可以知道多云的时候,无论气温、湿度和风力如何,都会外出;而下雨和晴朗的时候,会根据气温、湿度和风力的情况再来决定是否外出,因此决策树的构造变成了:

image

而下雨和晴朗的时候,会根据气温、湿度和风力的情况再来决定是否外出,这时就需要计算:

下雨时,选择信息增益最大的属性(气温、湿度或风力)来作为下一个节点
晴朗时,选择信息增益最大的属性(气温、湿度或风力)来作为下一个节点

天气=下雨的情况下:

由前面知道H(X=外出,Y=下雨)=0.971,在下雨时气温、湿度和风力外出情况如下所示:

天气 气温 湿度 风力 外出
下雨 温暖 无风
下雨 寒冷 正常 无风
下雨 寒冷 正常 有风
下雨 温暖 正常 无风
下雨 温暖 有风

气温的信息熵H(X_1=外出,X_2=下雨|Y=气温)为:

\begin{aligned} H(X_1=外出,X_2=下雨|Y=气温)&=H(X_1=外出,X_2=下雨|Y=温暖)+H(X_1=外出,X_2=下雨|Y=寒冷) \\ &=\frac{3}{5}\times (\frac{2}{3}log_2\frac{1}{\frac{2}{3}}+\frac{1}{3}log_2\frac{1}{\frac{1}{3}})+\frac{2}{5}\times (\frac{1}{2}log_2\frac{1}{\frac{1}{2}}+\frac{1}{2}log_2\frac{1}{\frac{1}{2}}) \\ &=0.951 \end{aligned}

同理可得 湿度的信息熵H(X_1=外出,X_2=下雨|Y=湿度)=0.951风力的信息熵H(X_1=外出,X_2=下雨|Y=风力)=0

天气=晴朗的情况下:

H(X_1=外出,X_2=晴朗|Y=湿度)最大,那么新得到的树形如下所示:

image

而上面却发现没有气温选项?因为在其他三个的条件下就能决定是否外出了,气温对于外出没有直接影响。

计算代码

将文本转化为数字:

data[data == u'晴朗'] = 2
data[data == u'多云'] = 1
data[data == u'下雨'] = 0
data[data == u'高温'] = 2
data[data == u'温暖'] = 1
data[data == u'寒冷'] = 0
data[data == u'高'] = 1
data[data == u'正常'] = 0
data[data == u'有风'] = 1
data[data == u'无风'] = 0
data[data == u'是'] = 1
data[data == u'否'] = 0

基于信息熵来计算:

# 建立决策树模型,基于信息熵
dtc = DTC(criterion='entropy')
dtc.fit(x_train, y_train)

预测:

print(dtc.predict(x_test))
print(dtc.score(x_test, y_test))

这里样本比较小,所以得到的准确率分数为1.0

image

数据和代码下载请关注公众号【 机器学习和大数据挖掘 】,后台回复【 机器学习 】即可获取

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 201,681评论 5 474
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 84,710评论 2 377
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 148,623评论 0 334
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,202评论 1 272
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,232评论 5 363
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,368评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,795评论 3 393
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,461评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,647评论 1 295
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,476评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,525评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,226评论 3 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,785评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,857评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,090评论 1 258
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,647评论 2 348
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,215评论 2 341