Розбиття дерева
Маючи індекс Джині, який утвориться під час розбиття за змінною A або B відповідно, визначте, за якою змінною дерево має розбиватися далі.
Ця вправа є частиною курсу
HR Analytics: прогнозування плинності персоналу в 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!")