트리 분할하기
변수 A 또는 B로 분할했을 때의 Gini 지수가 각각 주어졌습니다. 다음 분할은 어느 변수를 기준으로 해야 할지 결정하세요.
이 연습은 강의의 일부입니다
HR Analytics: Python으로 직원 이탈 예측하기
연습 안내
- 어떤 Gini 값이 더 낮은지 확인하세요.
- A의 Gini가 더 낮으면 A로 분할하고, 그렇지 않으면 B로 분할하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Gini index in case of splitting by variable A or B
gini_A = 0.65
gini_B = 0.15
# check which Gini is lower and use it for spliting
if ____ < ____:
print("split by A!")
____:
print("split by B!")