Gini-Index berechnen
Der Decision-Tree-Algorithmus zielt darauf ab, in den Endknoten möglichst reine Partitionen zu erreichen. Der Gini-Index ist eine Methode, um das zu messen. Er wird auf Basis des Anteils der Stichproben in jeder Gruppe berechnet.
Gegeben sind die Anzahl der Personen, die geblieben sind, und derjenigen, die gegangen sind. Berechne den Gini-Index für diesen Knoten.
Diese Übung ist Teil des Kurses
HR Analytics: Mitarbeiterfluktuation mit Python vorhersagen
Anleitung zur Übung
- Berechne die Gesamtzahl der Mitarbeitenden in diesem Knoten.
- Berechne den Gini-Index basierend auf dem Anteil der Mitarbeitenden in jeder Gruppe.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)