计算 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)