Gini 지수 계산하기
Decision Tree 알고리즘은 말단 노드의 데이터가 가능한 한 순수해지도록 분할하는 것을 목표로 해요. 이를 위해 사용하는 방법 중 하나가 Gini 지수이며, 각 그룹에 속한 표본의 비율을 바탕으로 계산합니다.
각각 남은 인원(stayed)과 이직한 인원(left)의 수가 주어졌을 때, 해당 노드의 Gini 지수를 계산하세요.
이 연습은 강의의 일부입니다
HR Analytics: Python으로 직원 이탈 예측하기
연습 안내
- 해당 노드의 전체 직원 수를 계산하세요.
- 각 그룹에 속한 직원 비율을 바탕으로 Gini 지수를 구하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
#number of people who stayed/left
stayed = 37
left = 1138
#sum of stayed and left
total = ____ + ____
#gini index
gini = ____*(____/total)*(____/total)