新知一下
海量新知
6 0 7 1 4 1 4

CART决策树算法【AI学习】

IT技术订阅 | IT技术全面介绍 2020/08/05 10:05

阅读目录

  • 1 引言

  • 2 分类问题

  • 3 回归问题

  • 4 树剪枝

  • 5 总结

注:本系列所有博客将持续更新并发布在github和gitee上,您可以通过github、gitee下载本系列所有文章笔记文件。

1 引言

上一篇博客中介绍了ID3和C4.5两种决策树算法,这两种决策树都只能用于分类问题,而本文要说的CART(classification and regression tree)决策树不仅能用于分类问题,也能用于回归问题。

与ID3算法和C4.5算法相比,CART 还有个特性就是其所有非叶子结点都只有两个子树,也就是说在根据特征属性分裂数据集时,无论该特征属性有多少个可能取值,都只有两种选择——‘是’和‘否’,以上文中判断是否是程序员数据集为例,如果根据近视程度进行分裂,可以将数据集分为{‘轻微’}和{‘中等’,‘严重’}两个数据集(当然也可以是其两种组合)然后在进一步迭代中进一步细化分数据集。

下面,我们分别说说CART算法如何解决分类问题和回归问题。

2 分类问题

对于分类问题,CART算法采用基尼指数作为最优分裂特征属性选择标准。

先来说说基尼指数,与熵一样,基尼指数越小则数据集不确定性越小,代表着数据集纯度越高。给定数据集

X

X

包含

L

L

个分类,那么数据集

X

X

的基尼指数为:

G i n i ( X ) = l L | X l | | X | ( 1 | X l | | X | ) = 1 l = 1 L ( | X l | | X | ) 2

G

i

n

i

(

X

)

=

l

L

|

X

l

|

|

X

|

(

1

|

X

l

|

|

X

|

)

=

1

l

=

1

L

(

|

X

l

|

|

X

|

)

2

假设

A

A

是数据集

X

X

中包含若干个可能取值的一个特征属性,

a

a

A

A

的其中一个可能取值,将数据集

X

X

按照

a

a

进行划分,就可以分为两个数据集,分别是

X 1 = { x X | x A = a }

X

1

=

{

x

X

|

x

A

=

a

}

X 2 = { x X | x A a }

X

2

=

{

x

X

|

x

A

a

}

,那么在特征

A

A

下,集合

X

X

的基尼指数为:

G i n i ( X , A ) = X 1 X G i n i ( X 1 ) + X 2 X G i n i ( X 2 )

G

i

n

i

(

X

,

A

)

=

|

X

1

X

|

G

i

n

i

(

X

1

)

+

|

X

2

X

|

G

i

n

i

(

X

2

)

接下来,我们通过实例演示如何应用基尼指数选择最优分裂特征属性。还是使用上篇博客中介绍ID3算法时使用过的数据集,如下所示。先来计算三个特征属性各个可能取值的基尼指数。

新知达人, CART决策树算法【AI学习】

对属性

A

A

的“穿格子衬衫”这个值计算基尼指数:

G i n i ( X , A 1 ) = 5 10 × { 2 × 4 5 × 1 5 } + 5 10 × { 2 × 3 5 × 2 5 } = 0.4

G

i

n

i

(

X

,

A

1

)

=

5

10

×

{

2

×

4

5

×

1

5

}

+

5

10

×

{

2

×

3

5

×

2

5

}

=

0.4

对属性

A

A

的“不穿格子衬衫”这个值计算基尼指数,由于只有两个属性,无论按照哪个属性来计算结果都一样,所以:

G i n i ( X , A 2 ) = G i n i ( X , A 1 ) = 0.4

G

i

n

i

(

X

,

A

2

)

=

G

i

n

i

(

X

,

A

1

)

=

0.4

对属性

B

B

的“严重”这个值计算基尼指数:

G i n i ( X , B 1 ) = 3 10 × { 2 × 2 3 × 1 3 } + 7 10 × { 2 × 5 7 × 2 7 } = 0.42

G

i

n

i

(

X

,

B

1

)

=

3

10

×

{

2

×

2

3

×

1

3

}

+

7

10

×

{

2

×

5

7

×

2

7

}

=

0.42

对属性

B

B

的“中等”这个值计算基尼指数:

G i n i ( X , B 2 ) = 4 10 × { 2 × 4 4 × 0 4 } + 6 10 × { 2 × 3 6 × 3 6 } = 0.3

G

i

n

i

(

X

,

B

2

)

=

4

10

×

{

2

×

4

4

×

0

4

}

+

6

10

×

{

2

×

3

6

×

3

6

}

=

0.3

对属性

B

B

的“轻微”这个值计算基尼指数:

G i n i ( X , B 3 ) = 3 10 × { 2 × 1 3 × 2 3 } + 7 10 × { 2 × 6 7 × 1 7 } = 0.46

G

i

n

i

(

X

,

B

3

)

=

3

10

×

{

2

×

1

3

×

2

3

}

+

7

10

×

{

2

×

6

7

×

1

7

}

=

0.46

对属性

C

C

的“严重”这个值计算基尼指数:

G i n i ( X , C 1 ) = 3 10 × { 2 × 0 3 × 3 3 } + 7 10 × { 2 × 4 7 × 3 7 } = 0.34

G

i

n

i

(

X

,

C

1

)

=

3

10

×

{

2

×

0

3

×

3

3

}

+

7

10

×

{

2

×

4

7

×

3

7

}

=

0.34

对属性

C

C

的“中等”这个值计算基尼指数:

G i n i ( X , C 2 ) = 3 10 × { 2 × 1 3 × 2 3 } + 7 10 × { 2 × 5 7 × 2 7 } = 0.42

G

i

n

i

(

X

,

C

2

)

=

3

10

×

{

2

×

1

3

×

2

3

}

+

7

10

×

{

2

×

5

7

×

2

7

}

=

0.42

对属性

C

C

的“轻微”这个值计算基尼指数:

G i n i ( X , C 3 ) = 3 10 × { 2 × 1 3 × 2 3 } + 7 10 × { 2 × 6 7 × 1 7 } = 0.46

G

i

n

i

(

X

,

C

3

)

=

3

10

×

{

2

×

1

3

×

2

3

}

+

7

10

×

{

2

×

6

7

×

1

7

}

=

0.46

可见,属性

B

B

的“中等“取值时具有最小的基尼指数,所以这个值作为当前数据集的最优分裂特征属性值。分裂后,可以获得两个数据集,对获得的数据集继续计算基尼指数,选择最优分裂特征属性值,如此迭代形成一颗完整的决策树。

对于连续型特征属性,可以参照C4.5算法对连续型特征属性的处理方法,只不过在CART算法中是计算基尼指数。

回到顶部

3 回归问题

此时,我们研究的已经是回归问题了(关于回归与分类,在讨论线性回归算法的时候已经分析过,如果还不清楚,传送门走起)所以,请转变思路,对于任意一个

x X

x

X

,经过决策树后的输出

f ( x )

f

(

x

)

的可能取值已经不再像之前的分类决策树那样,

f ( x )

f

(

x

)

的取值只可能是在

X

X

中出现过的那几种取值,回归树最后的输出

f ( x )

f

(

x

)

可能是之前没有出现过的,甚至连可能值的个数都不固定。所以,对于回归树,首先解决的问题就是如何确定

f ( x )

f

(

x

)

的可能值。

对于数据集

X

X

,假设我们在其特征属性

A

A

上早上一个值

a

a

将数据集划分成两类:

X 1 = { x | x A a }

X

1

=

{

x

|

x

A

a

}

X 2 = { x | x A > a }

X

2

=

{

x

|

x

A

>

a

}

在这两个类上的输出值

f ( x )

f

(

x

)

分别为

c 1

c

1

c 2

c

2

,那么根据特征属性

A

A

的值

a

a

X

X

进行划分,所产生的总误差是:

L o s s A , a = x X 1 ( y c 1 ) 2 + x X 2 ( y c 2 ) 2

L

o

s

s

A

,

a

=

x

X

1

(

y

c

1

)

2

+

x

X

2

(

y

c

2

)

2

式中,

y

y

x

x

对应的真实值。我们的目标就是使得

L o s s A , a

L

o

s

s

A

,

a

最小化时的

c 1

c

1

c 2

c

2

,目标函数为:

min x X 1 ( y c 1 ) 2 + min x X 2 ( y c 2 ) 2

min

x

X

1

(

y

c

1

)

2

+

min

x

X

2

(

y

c

2

)

2

那么,当

c 1

c

1

c 2

c

2

取什么值的的时候

L o s s A , a

L

o

s

s

A

,

a

最小呢?根据最小二乘的性质可知,当

c 1

c

1

c 2

c

2

分为为

X 1

X

1

X 2

X

2

中所有

y

y

的平均值的时候

c 1

c

1

c 2

c

2

去的最小值,即:

c i = a v e ( y | x X i )

c

i

=

a

v

e

(

y

|

x

X

i

)

所以,如果根据

a

a

划分之后得到的是叶子结点,那么最终输出的值就是所属样本子集所有

y

y

的平均值。

f ( x ) = c i = a v e ( y | x X i )

f

(

x

)

=

c

i

=

a

v

e

(

y

|

x

X

i

)

对数如何确定输出值的问题,就已经解决了。接下来还剩两个个问题需要解决,那就是选择哪个属性作为最优分割特征属性以及选择哪个值作为最佳的分割点。

对于这个问题,可以通过遍历数据集各特征属性的可能取值的方式来解决:对数据集

X

X

中各特征属性

A

A

,计算其所有取值

a

a

下的

L o s s A , a

L

o

s

s

A

,

a

,然后对比所有

L o s s A , a

L

o

s

s

A

,

a

,取值最小的

L o s s A , a

L

o

s

s

A

,

a

所对应的特征属性

A

A

为当前最优分裂特征属性,

a

a

为最佳分裂点。

至此,如何确定各分支的输出值、如何选择最优分割特征属性和分割点的问题都已解决,最后总结一下CART算法在回归问题中的决策树构建流程:

(1)对当前数据集

X

X

,计算所有特征属性

A

A

下所有取值

a

a

作为分割点时的最小

L o s s A , a

L

o

s

s

A

,

a

(2)对比所有

L o s s A , a

L

o

s

s

A

,

a

,选择最小的

L o s s A , a

L

o

s

s

A

,

a

所对应的特征属性

A

A

为当前最优分裂特征属性,

a

a

为最佳分裂点将数据集划分都左右两个子树中;

(3)对左右两个子树的数据集重复(1)、(2)步骤继续划分,直到节点中数据集满足指定条件则决策树构建完成。

回到顶部

4 树剪枝

无论是面对分类问题,还是回归问题,最终生成的树都有可能过于复杂,容易发生过拟合的情况,所以决策树构建完成后,有必要进一步完成数剪枝。

本文代价复杂度剪枝 Cost-Complexity Pruning(CCP) 方法,过程如下:

输入:CART算法生成的决策树

T 0

T

0

输出:剪枝后的最优决策树

T α

T

α

(1)令

k = 0

k

=

0

T = T 0

T

=

T

0

α = +

α

=

+

(2)自上而下地对各内部节点计算

C ( T t )

C

(

T

t

)

| T t |

|

T

t

|

以及

g ( t ) = C ( t ) C ( T t ) | T t | 1

g

(

t

)

=

C

(

t

)

C

(

T

t

)

|

T

t

|

1

α = min ( α , g ( t ) )

α

=

min

(

α

,

g

(

t

)

)

其中,

T t

T

t

表示以

t

t

为根节点的子树,

C ( t )

C

(

t

)

是对

t

t

进行剪枝后对训练数据集的预测误差,

C ( T t )

C

(

T

t

)

是对训练数据集的预测误差,

| T t |

|

T

t

|

T t

T

t

的叶子结点个数;

(3)自上而下地访问内部节点

t

t

,如果有

g ( t ) = α

g

(

t

)

=

α

,则对

t

t

进行剪枝,并对叶子结点

t

t

以多数表决法决定输出,得到树

T

T

(4)令

k = k + 1

k

=

k

+

1

α k = α

α

k

=

α

T k = T

T

k

=

T

(5)如果

T

T

不是由根节点单独构成的树,则回到步骤(3);

(6)采用交叉验证法在子树序列

T 0 , T 1 , , T k = T

T

0

,

T

1

,

,

T

k

=

T

选取最优的子树

T α

T

α

要理解CART决策树的整个剪枝过程,关键是明白

g ( t )

g

(

t

)

的含义,对于一颗理想的决策树,我们当然希望预测误差越小越好,树的规模也越小越好,但是两者却不能两全,因为往往预测误差随着树规模的增大而减小,所以单独考虑预测误差变化或者树规模变化都不合适,最好是选择一个衡量标准能够同时考虑到预测误差变化量和树规模变化,例如两者的比值。

仔细

g ( t )

g

(

t

)

的计算发现,分子是剪枝前后预测误差相减,也就是预测误差变化量,分子是剪枝前后叶子结点数的变化量,所以我们可以认为两者的比值就是树

t

t

每个叶子节点所带来的的预测误差的变化量,或者说树

t

t

所带来的的预测误差变化率——这就是

g ( t )

g

(

t

)

的含义。

为什么每次对

g ( t )

g

(

t

)

最小的节点进行剪枝呢?因为

g ( t )

g

(

t

)

越小代表对

t

t

对整个决策树的作用越小,对其进行剪枝对决策树的准确率影响也是最想的,当然应该优先被剪枝。

如果还不明白,那么通过下面的例子来理解吧(例子来源于:https://www.jianshu.com/p/b90a9ce05b28)。

有下面这个坐标中中的数据集,以及根据数据集构建好的决策树,如下图所示:

新知达人, CART决策树算法【AI学习】

此时,

α 1 = 0

α

1

=

0

,树中共有3个节点,对每个节点分别计算其

g ( t )

g

(

t

)

新知达人, CART决策树算法【AI学习】

t 1

t

1

t 2

t

2

节点的

g ( t )

g

(

t

)

最小,我们选择剪枝少的节点,也就是

t 3

t

3

进行剪枝,并且令

α 2 = 1 8

α

2

=

1

8

。剪枝后决策树如下:

新知达人, CART决策树算法【AI学习】

剩下两个节点,继续计算每一个节点的

g ( t )

g

(

t

)

:

新知达人, CART决策树算法【AI学习】

显然,

t 2

t

2

g ( t )

g

(

t

)

更小,所以对

t 2

t

2

进行剪枝,并令

α 3 = 1 8

α

3

=

1

8

新知达人, CART决策树算法【AI学习】

这时候仅剩下一个

t 1

t

1

,计算后有

g ( t 3 ) = 1 4

g

(

t

3

)

=

1

4

,所以

α 4 = 1 4

α

4

=

1

4

完成上述所有计算后,我们得到序列

α 0 = 0 , α 2 = 1 8 , α 3 = 1 8 , α 4 = 1 4

α

0

=

0

,

α

2

=

1

8

,

α

3

=

1

8

,

α

4

=

1

4

,以及对应的子树。接下来剩余的工作就是利用独立的验证数据集计算每个子树的平方误差或者基尼指数,选择误差最小的那个子树作为最优的剪枝后的树。

回到顶部

5 总结

在本文末尾对3种决策树算法做一个简单对比总结:

新知达人, CART决策树算法【AI学习】


更多“算法”相关内容

更多“算法”相关内容

新知精选

更多新知精选