Вычисление индекса Джини
Алгоритм дерева решений стремится к тому, чтобы разбиения в конечных узлах были как можно более однородными. Один из способов достичь этого — индекс Джини. Он вычисляется на основе доли образцов в каждой группе.
По заданному числу сотрудников, которые остались и уволились, вычислите индекс Джини для этого узла.
Это упражнение является частью курса
HR-аналитика: прогнозирование текучести кадров на Python
Инструкции к упражнению
- Вычислите общее количество сотрудников в этом узле.
- Рассчитайте индекс Джини на основе доли сотрудников в каждой группе.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)