最小二乘回归树生成算法

来源:互联网 发布:域名访问升级访问 编辑:程序博客网 时间:2024/05/16 01:17

算法:

输入:训练数据集 D
输出:回归树 f(x) .
在训练数据集所在的输入空间中,递归地将每个区域划分为两个子区域并决定每个子区域上的输出值,构建二叉决策树:
(1) 选择最优切分变量 j 与切分点 s,求解

minj,sminc1xiR1(j,s)(yic1)2+minc2xiR2(j,s)(yic2)2
遍历变量 j,对固定的切分变量 j 扫描切分点 s
(2) 用选定的对 (j,s) 划分区域并决定相应的输出值:
R1(j,s)={x|x(j)s},R2(j,s)={x|x(j)>s}
c^m=1NmxiRm(j,s)yi,xRm,m=1,2

(3) 继续对两个子区域调用步骤(1),(2),直至满足停止条件。
(4) 将输入空间分为 M 个区域 R1,R2,,RM,生成决策树:
f(x)=m=1Mc^mI(xRm)

0 0
原创粉丝点击