Obliczanie indeksu Giniego
Algorytm drzewa decyzyjnego dąży do tworzenia jak najbardziej jednorodnych podziałów w węzłach terminalnych. Jedną z metod służących do osiągnięcia tego celu jest indeks Giniego. Oblicza się go na podstawie proporcji próbek w każdej grupie.
Mając dane dotyczące liczby pracowników, którzy zostali i którzy odeszli, oblicz indeks Giniego dla tego węzła.
To ćwiczenie jest częścią kursu
HR Analytics: Przewidywanie rotacji pracowników w Pythonie
Instrukcje do ćwiczenia
- Oblicz całkowitą liczbę pracowników w tym węźle.
- Oblicz indeks Giniego na podstawie proporcji pracowników w każdej grupie.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)