Fractionner l'arbre
Étant donné l'indice de Gini qui résulterait d'un fractionnement selon la variable A ou B, déterminez par quelle variable l'arbre devrait se fractionner ensuite.
Cette activité fait partie du cours
Analytique RH : prédire le roulement du personnel avec Python
Instructions de l’exercice
- Vérifiez quel indice de Gini est le plus faible.
- Si A a un indice de Gini plus faible, fractionnez selon A, sinon, fractionnez selon B.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# 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!")