《统计学习方法》极简笔记P5:决策树公式推导

决策树学习

通过训练数据集
为输入实例
为类标记
学习目标:对实例进行正确分类

特征选择



条件熵

信息增益的算法
输入:特征A,训练集D
输出:特征A对训练集D的信息增益
(1)计算训练集D的经验熵

(2)计算特征A对训练集D的经验条件熵

(3)计算信息增益

信息增益比的算法

其中

决策树剪枝

树T的叶结点个数为|T|,t是树T的叶结点,该叶结点有个样本点,其中k类的样本点有个,为叶结点t上的经验熵,则决策树学习的损失函数

其中


树的剪枝算法
输入:生成的决策树T,参数α
输出:修剪后的子树
(1)计算每个结点经验熵
(2)递归地从树的叶结点向上回溯,若:
则剪枝,其中为剪枝后,为剪枝前
(3)返回(2),直至得到损失函数最小的子树

CART算法--回归树

对输入空间划分为M个单元,每个划分上有固定输出,则回归树模型表示为

预测误差:
划分单元上的最优值
输入空间的划分采用启发式方法
选择第j个变量和它的取值s作为切分变量和切分点,定义两个区域:
,
然后寻找最优切分变量j和最优切分点s,求解:

对固定输入变量j,找到最优切分点s


最小二乘回归树生成算法
输入:训练集D
输出:回归树f(x)
(1)选择最优切分变量j和切分点s,求解

遍历j,对固定j扫描切分点s,找出使上式达到最小的(j,s)组合
(2)选择(j,s)划分区域并决定相应输出值:
,


(3)对两个子区域循环(1),(2),直至满足停止条件
(4)将输入空间划分为M个单元,生成决策树:

CART算法--分类树

分类树用基尼指数选择最优特征,同时决定该特征的最优二值切分点
基尼指数
K个类,属于第k类的概率,则

样本集D的基尼指数

样本D根据特征A是否取值a被划分为,


CART生产算法
输入:训练数据集D,停止计算的条件;
输出:CART决策树.
(1)训练数据集D,计算现有特征对该数据集的基尼指数.此时,对每一个特征A,对其可能取的每个值a,根据样本点对A=a的测试为“是”或“否”将D分割成D1和D2两部分,根据计算A=a时的基尼指数;
(2)在所有可能的特征A以及它们所有可能的切分点a中,选择基尼指数最小的特征及其对应的切分点作为最优特征与最优切分点.依最优特征与最优切分点,从现结点生成两个子结点,将训练数据集依特征分配到两个子结点中去;
(3)对两个子结点递归地调用(1)(2),直至满足停止条件.
(4)生成CART决策树.

CART剪枝算法

输入:CART算法生成的决策树;
输出:最优决策树.
(1)设.
(2)设
(3)自下而上地对各内部结点t计算以及
,
这里表示以t为根结点的子树.,是对训练数据的预测误差,引是的叶结点个数;
(4)对的内部结点t进行剪枝,并对叶结点t以多数表决法决定其类,得到树T;
(5)设;
(6)如果T不是由根结点单独构成的树。则回到步骤(4);
(7)采用交叉验证法在子树序列中选取最优子树

你可能感兴趣的:(《统计学习方法》极简笔记P5:决策树公式推导)