인프런 커뮤니티 질문&답변

이진분류 f1에 평가를 할 때 predict를 사용합니다. 이 결과는 예를 들면 0또는1, A 또는 B인 1개 컬럼이고

predict_proba와 같이 2개 컬럼이 아니에요 따라서 pred[:,1] 와 같이 2개 컬럼 중 하나를 선택하지 않아도 됩니다.

1. f1_score 사용 시:

   • f1_score는 실제 레이블과 예측 레이블을 비교하여 계산됩니다. 이 때, pos_label 파라미터를 사용하여 어느 레이블을 양성으로 간주할지 지정합니다.

      

2. roc_auc_score 사용 시:

   • roc_auc_score는 실제 레이블과 예측 확률 또는 점수를 비교하여 계산됩니다.

   • 이진 분류에서는 일반적으로 예측 확률의 두 번째 열(pred[:, 1])을 사용합니다. 이는 대부분의 이진 분류 모델에서 '1' 레이블(양성 클래스)에 대한 확률을 두 번째 열로 반환하기 때문입니다.