开始使用免费开始使用

划分决策树

已知按变量 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!")
编辑并运行代码