1. 이직할 직원 예측하기

1.1 개요

• 요즘 회사들의 큰 고민 중 하나는 직원들의 이직이다.
• 직원이 조만간 이직을 할지 안 할지를 예측하는 모델을 만들어 보자.
• 다음과 같은 과정을 수행한다.
  • 우선 기존 직원들의 정보(나이, 성별, 출신학교, 전공, 부서, 연봉 등 회사에서 수집 가능한 모든 데이터)를 수집한다.
  • 그 결과 현재 그 직원이 이직했는지 안 했는지를 표시한다.
  • 그 다음 직원 정보와 이직 유무라는 결과 사이에 어떤 관계가 있는지 인공지능에서 학습시킨다.

 

대표적인 모델 4개 선택

• Neural Network
• Random Forest
• SVM
• Naive Bayes

• 학습이 완료되면 Test and Score를 더블클릭해서 모델의 테스트 결과 즉 성능을 확인한다.

• Recall에서 마우스의 오른쪽 버튼을 눌러 LogLoss도 선택한다.

[1.5 Confusion Matrix]

 

1.6 예측

• '객관식예측_이직할직원_적용' 파일을 불러오자.
• '이직' 항목의 Type을 Categorical로 바꾸고, Role은 Tartget으로 변경 한다.

  

• 'Evaluate'에서 'Prediction'을 클릭한 후 오른쪽 화면에 불러온다.
  
  

• Prediction을 더블클릭하여 예측 결과를 확인한다.

• 모델들 마다 예측 결과가 다른 경우 좌측 상단의 Show Probablity for는 Yes No 각각의 확률을 몇 % 인지 알려준다.