CART 剪枝算法

来源:互联网 发布:6级野猪骑士升级数据 编辑:程序博客网 时间:2024/05/18 17:01

算法:

输入:CART 算法生成的决策树 T0
输出:最优决策树 Tα .
(1) 设 k=0,T=T0 .
(2) 设 α=+ .
(3) 自下而上地对各内部结点 t 计算 C(Tt)|Tt|以及

g(t)=C(t)C(Tt)|Tt|1
α=min(α,g(t))
,这里 Tt 表示以 t 为结点的子树,C(Tt) 是对训练数据的误差估计,|Tt|Tt 的叶结点个数。
(4) 自上而下地访问内部结点 t,如果有 g(t)=α,进行剪枝,并对叶结点 t 以多数表决法决定其类,得到树 T .
(5) 设 k=k+1,αk=α,Tk=T .
(6) 如果 T 不是由根结点单独构成的树,则返回步骤(4) .
(7) 采用交叉验证法在子树序列 T0,T1,,Tn 中选取最优子树 Tα .

0 0
原创粉丝点击