Обчислення індексу Джині
Алгоритм дерева рішень прагне отримати розбиття у кінцевих вузлах якомога чистішими. Один зі способів це досягти — індекс Джині. Його обчислюють на основі частки вибірок у кожній групі.
Маючи кількість людей, які відповідно залишилися та пішли, обчисліть індекс Джині для цього вузла.
Ця вправа є частиною курсу
HR Analytics: прогнозування плинності персоналу в Python
Інструкції до вправи
- Обчисліть загальну кількість співробітників у цьому вузлі.
- Обчисліть індекс Джині на основі частки співробітників у кожній групі.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)