1. 학습
    2. /
  2. 강의
    3. /
  3. HR Analytics: Python으로 직원 이탈 예측하기
Connected

연습 문제

Gini 지수 계산하기

Decision Tree 알고리즘은 말단 노드의 데이터가 가능한 한 순수해지도록 분할하는 것을 목표로 해요. 이를 위해 사용하는 방법 중 하나가 Gini 지수이며, 각 그룹에 속한 표본의 비율을 바탕으로 계산합니다.

각각 남은 인원(stayed)과 이직한 인원(left)의 수가 주어졌을 때, 해당 노드의 Gini 지수를 계산하세요.

지침

100 XP
  • 해당 노드의 전체 직원 수를 계산하세요.
  • 각 그룹에 속한 직원 비율을 바탕으로 Gini 지수를 구하세요.