决策树中的ID3算法(过程详解)

背景知识:

决策树是一种类似于流程图的树结构模型,它通过对数据特征的一系列测试和划分来进行决策或预测。

信息熵

信息熵体现了数据集中的不确定性程度。

●当信息熵较高时,表明数据中的不确定性大,预测难度增加

●信息熵较低则意味着不确定性小,数据更易于预测。

示例:考虑贷款申请数据集,只有批准和拒绝两个类,其中9个批准,6个拒绝。

条件熵:

条件熵表示在某属性A 的取值已知的情况下,数据集的不确定性:

信息增益量

信息增益量化了选择某个特征进行划分后数据集纯度的提升信息增益越大,该特征越适合作为决策树的分割条件。

衡量某一属性对数据集分类的效果,表示使用该属性后数据集不确定性的减少量**】**

Gain(D,A) 越大,属性 A 对分类的贡献越大。

分类后,批准和拒绝两种情况的概率

前面是3个部分的加权平均

ID3算法

ID3算法是一种基于信息增益来选定决策树节点特征的算法。

通过不断选择信息增益最大的属性,将数据集划分为多个子集,最终构建决策树。

为了计算方便,写的时候可以分开写

计算数据集熵 H(D):

计算特征"Outlook"的信息增益:

计算每个子集 Dv的熵 H(Dv)。

Outlook分位sunny,rainy,overcast三点

计算每个子集的加权熵,并求出总熵

计算信息增益:用数据集熵减去总熵。

就这样计算每个特征,比较各特征的信息增益,选择信息增益最高的特征作为当前节点的划分依据。


ID3算法 是一种基于信息增益的决策树生成算法。以下总结其解决计算题的思路,并通过一个示例详细展示其计算过程。


ID3算法计算思路

  1. 根据给定的样本数据,构造决策树,找到使信息增益最大的划分属性。

  2. 步骤概述

    • 计算数据集的信息熵(Entropy)
    • 对每个属性,计算基于该属性划分的数据的条件熵(Conditional Entropy)
    • 计算信息增益: 信息增益=数据集熵−条件熵信息增益 = 数据集熵 - 条件熵信息增益=数据集熵−条件熵
    • 选择信息增益最大的属性作为当前节点的划分属性。
    • 递归进行划分,直至满足停止条件(如所有样本属于同一类,或属性集为空)。
相关推荐
大草原的小灰灰几秒前
C++ STL之容器介绍(vector、list、set、map)
数据结构·c++·算法
小刘|35 分钟前
数据结构的插入与删除
java·数据结构·算法
廖显东-ShirDon 讲编程1 小时前
《零基础Go语言算法实战》【题目 2-22】Go 调度器优先调度问题
算法·程序员·go语言·web编程·go web
HyperAI超神经1 小时前
微软与腾讯技术交锋,TRELLIS引领3D生成领域多格式支持新方向
人工智能·深度学习·机器学习·计算机视觉·3d·大模型·数据集
Lulsj2 小时前
代码随想录day24 | 贪心算法理论基础 leetcode 455.分发饼干 376.摆动序列 53. 最大子序和
算法·leetcode·贪心算法
KuaCpp2 小时前
算法初学者(图的存储)链式前向星
c++·算法
Cosmoshhhyyy2 小时前
LeetCode:2270. 分割数组的方案数(遍历 Java)
java·算法·leetcode
Nil_cxc2 小时前
机器学习周报-GNN模型学习
人工智能·学习·机器学习
打不了嗝 ᥬ᭄3 小时前
P3884 [JLOI2009] 二叉树问题
数据结构·算法·蓝桥杯
KeyPan4 小时前
【Ubuntu与Linux操作系统:一、Ubuntu安装与基本使用】
linux·运维·服务器·人工智能·深度学习·ubuntu·机器学习