Sustie

3种决策树：ID3、C4.5和CART

假设有以下数据集，记录着一些人的种族、收入、是否有子女以及他们是否有购买保险：

我们希望构造一棵决策树，来根据种族、收入、是否有子女这三个属性来预测一个人是否有购买保险。构造决策树的关键问题在于决定分叉处要根据哪个属性来进行分类。常见的选择方法有 3 种：ID3、C4.5 和 CART。

在下文中，将属性集$\{\text{Race}, \text{Income}, \text{Child}\}$记作$D$。将要预测的目标 Insurance 记作$T$。

ID3

ID3 选择属性的方法是：

\argmax_{X\in D} H(T) - H(T|X) = \argmin_{X\in D} H(T|X)

C4.5

C4.5 选择属性的方法是：

\argmax_{X\in D} \frac{H(T) - H(T|X)}{H(X)}

CART

CART 选择属性的方法是：

\argmax_{X \in D} G(T) - G(T|X) = \argmin_{X \in D} G(T|X)

其中$G(X)$是 Gini 指数，其定义为：

G(X) = 1- \sum_{x} p^2(x)

例如$G(T) = 1-(\frac{1}{2})^2-(\frac{1}{2})^2=\frac{1}{2}$。

条件 Gini 指数的定义和条件信息熵类似：

G(X|Y)=\sum_y p(y)G(X|Y=y)