基尼指数的计算

Posted 是璇子鸭

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了基尼指数的计算相关的知识,希望对你有一定的参考价值。

一、基尼指数的概念

表示样本集合中一个随机选中的样本被分错的概率

注:Gini指数越小表示集合被选中的样本被参错的概率越小,即集合的纯度越高。反之,集合越不纯。当集合中所有样本为一个类时,基尼指数为0。

二、计算公式

在这里插入图片描述

三、示例

考虑如下数据集:
在这里插入图片描述

(1)计算整个数据集的 Gini 指标值

易统计:
在这里插入图片描述
故带入公式得
在这里插入图片描述

(2)计算属性性别的 Gini 指标值

在这里插入图片描述
按属性竖着看计算,有:
在这里插入图片描述
在这里插入图片描述

以上是关于基尼指数的计算的主要内容,如果未能解决你的问题,请参考以下文章

统计学习笔记之决策树

B-经济学-基尼指数

使用基尼指数的决策树

如何获取决策树中的所有基尼指数?

CART决策树----基尼指数划分

为啥基尼指数考虑决策树中的二分法?