predict_proba vs predit

Sooin Yoon·2025년 6월 16일
  1. predict_proba
  • 확률(probability)를 반환
  • 이진 분류일 경우, 두 개의 확률을 반환: 클래스 0일 확률/ 클래스 1일 확률
  • pred = model.predict_proba(x)
  • 확률기반평가 : rou_auc_score, log_loss와 같은 평가 지표에 사용됨
  • pred[:, 1]을 사용하여 양성 클래스(1)의 확률만 추출 가능한 것
  1. predict
  • 클래스 자체(0 또는 1)을 반환
  • predict_proba의 결과에서 확률이 더 높은 쪽을 선택한 것과 같음
  • model.predict(x)
  • 최종 분류 결정

0개의 댓글