Výpočet Gini indexu
Algoritmus rozhodovacího stromu se snaží vytvářet co nejčistší oddíly v koncových uzlech. Jednou z metod, která to umožňuje, je Gini index. Počítá se na základě podílu vzorků v každé skupině.
Na základě počtu zaměstnanců, kteří zůstali, a těch, kteří odešli, vypočítej Gini index pro daný uzel.
Toto cvičení je součástí kurzu
HR Analytics: Predicting Employee Churn in Python
Pokyny k cvičení
- Vypočítej celkový počet zaměstnanců v daném uzlu.
- Vypočítej Gini index na základě podílu zaměstnanců v každé skupině.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)