Découper l’arbre
Étant donné l’indice de Gini obtenu en divisant selon la variable A ou B, décidez par quelle variable l’arbre doit se diviser ensuite.
Cet exercice fait partie du cours
Analytique RH : prédire l’attrition des employés en Python
Instructions
- Vérifiez quel Gini est le plus faible.
- Si A a un Gini plus faible, divisez selon A ; sinon, divisez selon B.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# 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!")