划分决策树
已知按变量 A 或变量 B 进行划分各自得到的 Gini 指数,请判断下一步应按哪个变量来划分决策树。
本练习是课程的一部分
HR Analytics:用 Python 预测员工流失
练习说明
- 检查哪个 Gini 更低。
- 如果 A 的 Gini 更低,则按 A 划分;否则按 B 划分。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Gini index in case of splitting by variable A or B
gini_A = 0.65
gini_B = 0.15
# check which Gini is lower and use it for spliting
if ____ < ____:
print("split by A!")
____:
print("split by B!")