决策树分裂时的特征选择
作者:互联网
类别特征
1. 信息增益
信息增益直观理解:在选定特征对数据进行划分后,数据分布不确定性减少的程度,信息增益越大的特征越好。
信息增益的缺点
信息增益倾向于选择类别数较多的特征
这怎么解释呢?从信息增益的计算公式可以看出,不管是选择什么特征,H(D) 项表述的是原数据分布的熵,是一样的,那我们只需要分析 H(D|A) 这一项就可以了,特征类别越多, H(D|A) 会越小吗?
假设总样本数目为100, A1 特征有2个类别, A2 特征有10个类别,并且每个类别内的数据都是01均匀分布,那么:
H(D|A1) = 1/2 * (H(D, A1=1) + H(D, A1=2))
H(D|A2) = 1/10 * (H(D, A2=1) + H(D, A2=2) + ... + H(D, A2=10))
可以看出,如果落到每个特征类别上的分布一样的话,这两种特征的条件墒是一样的。
但如果是极端情况,特征多的类别,每个类别只有一个样本,那么其条件墒是0 (墒是大于等于0的数),最小。
在现实中,我们的样本数量往往是有限的,类别越多的特征,分到每个特征类别下的样本会越少,那么出现样本类别不平衡的可能性越大,即条件墒越少,信息增益越大
2. 信息增益率
标签:特征选择,样本,增益,A1,特征,分裂,A2,类别,决策树 来源: https://www.cnblogs.com/xumaomao/p/15084942.html