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
<Kurs>HR Analytics: Mitarbeiterfluktuation mit Python vorhersagen</Kurs>Übungsanweisungen
- Berechne die Gesamtzahl der Mitarbeitenden in diesem Knoten.
- Berechne den Gini-Index basierend auf dem Anteil der Mitarbeitenden in jeder Gruppe.
Interaktive praktische Übung
Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)