Calcularea indicelui Gini
Algoritmul arborelui de decizie urmărește să obțină partiții în nodurile terminale cât mai pure posibil. Indicele Gini este una dintre metodele folosite pentru a atinge acest obiectiv. Se calculează pe baza proporției eșantioanelor din fiecare grup.
Dată fiind numărul de angajați care au rămas, respectiv care au plecat, calculează indicele Gini pentru acel nod.
Acest exercițiu face parte din cursul
HR Analytics: Predicția fluctuației angajaților în Python
Instrucțiuni pentru exercițiu
- Calculează numărul total de angajați din acel nod.
- Calculează indicele Gini pe baza proporției angajaților din fiecare grup.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)