Beräkna Gini-index
Beslutsträdsalgoritmen strävar efter att skapa partitioner i de terminala noderna som är så rena som möjligt. Gini-indexet är en av de metoder som används för att uppnå detta. Det beräknas utifrån andelen observationer i varje grupp.
Givet antalet anställda som stannade respektive slutade, beräkna Gini-indexet för den noden.
Den här övningen är en del av kursen
HR-analys: Förutsäg personalomsättning i Python
Övningsinstruktioner
- Beräkna det totala antalet anställda i den noden.
- Beräkna Gini-indexet baserat på andelen anställda i varje grupp.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)