开始使用免费开始使用

计算 Gini 指数

决策树算法希望终端节点(叶子节点)的划分尽可能纯。Gini 指数是实现这一目标的方法之一。它基于各组样本所占比例计算。

给定分别留下和离开的人数,计算该节点的 Gini 指数。

本练习是课程的一部分

HR Analytics:用 Python 预测员工流失

查看课程

练习说明

  • 计算该节点中的员工总数。
  • 根据各组员工所占比例计算 Gini 指数。

交互式实操练习

通过完成这段示例代码来试试这个练习。

#number of people who stayed/left
stayed = 37
left = 1138

#sum of stayed and left
total = ____ + ____

#gini index
gini = ____*(____/total)*(____/total)
编辑并运行代码