직원 이직 분류기 튜닝

HR Analytics: Python으로 직원 이탈 예측하기

Hrant Davtyan

Assistant Professor of Data Science American University of Armenia

과적합

과적합 존재:

훈련 정확도: 100%
테스트 정확도: 97.23%

해결 방법:

트리 최대 깊이 제한
리프 최소 샘플 수 제한

트리 가지치기

깊이 제한

model_depth_5 = DecisionTreeClassifier(
                max_depth=5, random_state=42)
# Train set Accuracy: 97.71%
# Test  set Accuracy: 97.06%

샘플 제한

model_sample_100 = DecisionTreeClassifier(
                   min_samples_leaf=100, random_state=42)
# Train set Accuracy: 96.58%
# Test set Accuracy: 96.13%

연습해 봅시다!

HR Analytics: Python으로 직원 이탈 예측하기