CART算法实例解析

CART算法的原理,网上一抓一大把,在这里我就不过多赘述,现在我直接用一个实例进行讲解,CATR算法在分裂时是如何计算的
注:CART算法一般适用于二分类,构建二叉树
1、数据集D:
年龄 有工作 有自己的房子 信贷情况 类别
青年 否 否 一般 否
青年 否 否 好 否
青年 是 否 好 是
青年 是 是 一般 是
青年 否 否 一般 否
中年 否 否 一般 否
中年 否 否 好 否
中年 是 是 好 是
中年 否 是 非常好 是
中年 否 是 非常好 是
老年 否 是 非常好 是
老年 否 是 好 是
老年 是 否 好 是
老年 是 否 非常好 是
老年 否 否 一般 否
2、开始计算
第一步,分别设四个特征(年龄、有工作、有自己的房子、信贷情况)为。四个特征的取值分别为(青年、中年、老年)(是、否),(是、否),(一般、好、非常好)
开始求取特征A1的基尼系数:

Gini(D,=青年) =

Gini(D,=中年) =

Gini(D,=老年) =
同理可得:
Gini(D,=是) = 0.32
Gini(D,=是) = 0.27
特征A4的基尼系数:
Gini(D,=1) = 0.36
Gini(D,=2) = 0.47
Gini(D,=3) = 0.32
四个特征中,Gini(D,=是) = 0.27最小,所以选择为最优特征,=是为最优且分点。此时根节点生成两个子节点,一个是叶子节点,另外一个是含有三个特征,对这个节点使用上述方法,得到=是为最优特征和最优切分点。按照上述步骤进行计算,所有节点都是叶子结点

  • 2
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值