시작하기무료로 시작하기

혼동 행렬 만들기

영상에서 보신 것처럼 혼동 행렬은 모델의 성능을 평가하는 데 사용돼요. 하지만 혼동 행렬을 만들기 전에, 예측된 확률 값을 컷오프를 사용해 1 또는 0으로 분류해야 해요.

참고: 1은 Inactive, 0은 Active를 의미합니다.

test_set의 모든 사례에 대해 이직 확률의 예측 값을 담은 prediction_test가 작업 공간에 준비되어 있어요.

이 연습은 강의의 일부입니다

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

강의 보기

연습 안내

  • prediction_test의 숫자 예측 값을 컷오프 0.5를 사용해 범주형 예측 벡터로 변환하세요.
  • prediction_categories와 테스트 세트의 실제 값(test_set$turnover)을 사용해 혼동 행렬을 만드세요.

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

# Classify predictions using a cut-off of 0.5
prediction_categories <- ___(prediction_test > 0.5, 1, 0)

# Construct a confusion matrix
conf_matrix <- ___(prediction_categories, test_set$turnover)
conf_matrix
코드 편집 및 실행