Inflearn Community Q&A

분류에서 roc-auc를 묻는다면 predict_proba를 사용해 주세요

predict 과 predict_proba는 서로 다른 유형의 예측 결과를 반환합니다

1. predict 메소드:

   • predict 메소드는 각 입력 샘플에 대한 예측된 클래스 레이블을 반환

   • 반환되는 값은 가장 높은 확률을 가진 클래스에 해당하는 레이블임

      

   • 예를 들어, 이진 분류에서는 0 또는 1의 클래스 레이블을 반환하며, 다중 클래스 분류에서는 해당하는 클래스의 레이블을 반환함

2. predict_proba 메소드:

   • predict_proba 메소드는 각 입력 샘플에 대해 모든 클래스의 예측 확률을 반환

   • 반환되는 값은 각 클래스에 속할 확률을 나타내는 0과 1 사이의 값으로, 각 샘플에 대해 클래스 수만큼의 확률이 반환

   • 예를 들어, 이진 분류에서는 각 샘플에 대해 [0.46, 0.56]과 같이 [클래스 0에 속할 확률, 클래스 1에 속할 확률]의 형태로 반환함