数据挖掘算法学习(六)CART算法

来源:互联网 发布:排课程表的软件 编辑:程序博客网 时间:2024/05/07 13:27

转载请附上链接http://blog.csdn.net/iemyxie/article/details/39520537

分类回归树算法:CART(Classification And Regression Tree)算法采用一种二分递归分割的技术,将当前的样本集分为两个子样本集,使得生成的的每个非叶子节点都有两个分支。因此,CART算法生成的决策树是结构简洁的二叉树。

分类树两个基本思想:第一个是将训练样本进行递归地划分自变量空间进行建树的想法,第二个想法是用验证数据进行剪枝。


CARTC4.5的不同之处是节点在分裂时使用GINI指数。GINI指标主要是度量数据划分或训练数据集D的不纯度为主,系数值的属性作为测试属性,GINI值越小,表明样本的纯净度越高(即该样本只属于同一类的概率越高)。选择该属性产生最小的GINI指标的子集作为它的分裂子集。

算法步骤:

CART_classification(DataSet,featureList, alpha,)

创建根节点R

如果当前DataSet中的数据的类别相同,则标记R的类别标记为该类

如果决策树高度大于alpha,则不再分解,标R的类别classify(DataSet)

递归情况:

标记R的类别classify(DataSet)

featureList中选择属性F(选择Gini(DataSet,F)最小的属性划分,连续属性参考C4.5的离散化过程(Gini作为划分标准)

根据F,将DataSet做二元划分DS_LDS_R

如果DS_LDS_R为空,则不再分解

如果DS_LDS_R都不为空,节点

    C_L= CART_classification(DS_L,featureList, alpha);

    C_R= CART_classification(DS_RfeatureList, alpha)

将节点C_LC_R添加为R的左右子节点


使用SQL实现核心代码:
 
rr:while (1=1) doset @weather = (select id from weather where class = 0 limit 0,1);set @feature =(select parent from finalgini where statetemp=1 limit 0,1);if (@weather is null ) thenleave rr;else if(@feature is null) thenupdate finalgini set statetemp = state; end if;end if;if (@weather is not null) thenb:beginset current_gini = (select min(gini) from finalgini where statetemp=1);set current_class = (select parent from finalgini where gini = current_gini);drop table if exists aa;create temporary table aa (namee varchar(100));insert into aa select class from finalgini where parent=current_class;insert into aa select class2 from finalgini where parent=current_class;tt:while (1=1) doset @x = (select namee from aa limit 0,1);if (@x is not null) thena0:begindrop table if exists bb;set @b=concat('create temporary table bb as \(select id from ', current_table,' where ',current_class,' regexp \'',@x,'\' and class = 0 \)');prepare stmt2 from @b;execute stmt2;set @count = (select count(distinct play) from bb left join weather on bb.id = weather.id); if (@count =1) thena1:beginupdate bb left join weather on bb.id=weather.id set class = current_num;set current_num = current_num+1;if (current_table ='cc') thendelete from cc where id in (select id from bb);end if;set @f=(select play from cc limit 0,1);if (@f is null) thenset current_table='weather';update finalgini set statetemp=state; end if;delete from aa where namee = @x;end a1;end if;if (@count>1) thenset @id = (select count(id) from bb); if(@id = 2) thenw:beginupdate bb left join weather on bb.id=weather.id set class = current_num where play='yes';set current_num = current_num+1;update bb left join weather on bb.id=weather.id set class = current_num where play='no';set current_num = current_num+1;if (current_table ='cc') thendelete from cc where id in (select id from bb);end if;set @f=(select play from cc limit 0,1);if (@f is null) thenset current_table='weather';update finalgini set statetemp=state; end if;delete from aa where namee = @x;end w;end if;if(@id > 2) then drop table if exists cc;create temporary table cc select * from weather inner join bb using(id);set current_table = 'cc';leave tt;end if;end if;if(@count=0) thendelete from aa where namee = @x; end if;end a0;else update finalgini set state=0 where parent=current_class;leave tt;end if;end while;update finalgini set statetemp=0 where parent=current_class;  end b;end if;end while;end |delimiter ;  

程序中表的解释:

2 classgini属性不同分类集合的gini
3finalgini存放各个属性的最优分类及对应gini

转载请附上链接http://blog.csdn.net/iemyxie/article/details/39520537

0 0