Разбиение дерева
Используя индексы Джини, полученные при разбиении по переменной A и по переменной B соответственно, определите, по какой из них дерево должно разветвиться следующим.
Это упражнение является частью курса
HR-аналитика: прогнозирование текучести кадров на Python
Инструкции к упражнению
- Проверьте, какой индекс Джини меньше.
- Если индекс Джини для A меньше, выполните разбиение по 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!")