[CART决策树]如何划分离散型特征

在CART(分类与回归树)算法中,当处理离散型属性或者类别型属性的时候,节点的分裂过程遵循以下步骤:

1. 生成所有可能的二分方式

  • 对于具有k个类别的离散属性,需枚举所有可能的非空真子集划分,将属性值分为两个互斥的子集(左子集和右子集)。

  • 由于对称性(左右子集互换不影响分裂效果),实际划分方式数为 2k−1−12k−1−1。

  • 例如,当属性有3个类别(红、蓝、绿)时,可能的有效划分包括:

    • {红} vs {蓝, 绿}
    • {蓝} vs {红, 绿}
    • {绿} vs {红, 蓝}
    • {红, 蓝} vs {绿}
    • {红, 绿} vs {蓝}
    • {蓝, 绿} vs {红}
      但通过对称性简化后,只需考虑前三种或后三种,总数为 23−1−1=323−1−1=3 种。

2. 计算每种划分的基尼指数

  • 对每个候选划分,计算分裂后左右子节点的基尼指数(Gini Index),并按样本占比加权求和。基尼指数公式为:

    Gini(D)=1−∑i=1Cpi2Gini(D)=1−i=1∑C​pi2​

    其中 CC 是类别数,pipi​ 是第 ii 类的比例。分裂后的总基尼指数为:

    Ginisplit=NleftNGini(Dleft)+NrightNGini(Dright)Ginisplit​=NNleft​​Gini(Dleft​)+NNright​​Gini(Dright​)

    选择使 GinisplitGinisplit​ 最小的划分方式。

3. 选择最优划分

  • 比较所有候选划分的基尼指数,选择下降最大(即总基尼指数最小,越小代表数据越纯)的划分作为分裂规则。例如,若将{红, 蓝}归为左子节点、{绿}归为右子节点时基尼指数最小,则分裂条件为"颜色是否为绿"。

4. 处理类别较多的情况

  • 当离散属性类别数 kk 较大时,穷举所有划分计算量过大。此时可采用优化策略:

    • 按目标变量排序:对类别按目标变量(如正类比例)排序,转化为类似连续变量的处理,仅需 k−1k−1 次划分。
    • 启发式方法:限制候选划分数量或使用贪心策略减少计算量。

示例

假设颜色属性为红、蓝、绿,样本分布如下:

  • 左子集候选划分 {红, 蓝}:包含红、蓝样本,基尼指数较低。
  • 右子集候选划分 {绿}:包含绿样本,基尼指数较低。
    若此划分的总基尼指数最小,则节点分裂为"颜色 ∈ {红, 蓝}"和"颜色 = 绿"。

结论

CART通过遍历离散属性的所有有效二分方式,选择基尼指数最小的划分,生成二叉树结构。

该方法确保局部最优性,同时通过优化策略应对高基数离散属性,平衡计算效率与分类性能。

相关推荐
PNP Robotics6 小时前
多伦多大学机器人峰会|物理AI与具身智能落地新趋势
人工智能·深度学习·机器学习·机器人
GIR1236 小时前
官方出品 | 多通道土壤呼吸测量系统市场现状与十五五规划深度报告:行业分析+趋势预测全收录
大数据·人工智能·机器学习
DFT计算杂谈10 小时前
无 Root 权限在 Tesla K80 零门槛部署 DeepSeek 大模型
linux·服务器·网络·数据库·机器学习
zhumin72611 小时前
舆情聚类与政府投诉联动分析:基于时间序列的社会治理框架
机器学习·数据挖掘·聚类
_Jimmy_12 小时前
AI应用开发工程师面试题库
人工智能·python·深度学习·机器学习·知识图谱
临床数据科学和人工智能兴趣组13 小时前
RStudio的Console(控制台)是一个非常重要的组件
人工智能·机器学习·数据分析·r语言·r语言-4.2.1
CIO_Alliance14 小时前
企业AI化转型如何用AI+iPaaS实现降本增效?
人工智能·低代码·机器学习·ai·ipaas·系统集成·企业级ai化转型
林泽毅15 小时前
PyTRIO快速入门(一):概念、推理与训练
人工智能·python·深度学习·机器学习·语言模型
陕西企来客16 小时前
2026年7月技术好GEO优化方案:算法适配与内容策略
人工智能·算法·机器学习·技术好geo优化
想会飞的蒲公英16 小时前
用 Streamlit 给文本分类模型做一个演示页面
人工智能·python·机器学习