Calcul de l'indice de Gini
L'algorithme de l'arbre de décision vise à obtenir des partitions dans les nœuds terminaux qui soient les plus pures possible. L'indice de Gini est l'une des méthodes utilisées pour y parvenir. Il est calculé à partir de la proportion d'échantillons dans chaque groupe.
Étant donné le nombre de personnes qui sont restées et celles qui sont parties, calculez l'indice de Gini pour ce nœud.
Cette activité fait partie du cours
Analytique RH : prédire le roulement du personnel avec Python
Instructions de l’exercice
- Calculez le nombre total d'employés dans ce nœud.
- Calculez l'indice de Gini à partir de la proportion d'employés dans chaque groupe.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)