해결된 질문
작성
·
443
답변 1
0
문제에서 확률을 묻거나 평가지표가 roc-auc인 경우에는 predict_proba를 사용합니다.
참고!
predict
와 predict_proba
는 분류 모델에서 사용되는 두 가지 다른 유형의 함수입니다. 이들의 주된 차이점은 반환하는 결과의 형태에 있습니다.
predict
함수사용: predict
함수는 각 데이터 포인트에 대한 예측된 클래스 레이블을 반환합니다.
결과: 이진 분류 문제에서 결과는 0 또는 1과 같은 레이블입니다. 다중 클래스 분류에서는 각 샘플에 대해 하나의 클래스 레이블을 반환합니다.
predict_proba
함수사용: predict_proba
는 각 클래스에 대한 예측 확률을 반환합니다.
결과: 이진 분류에서는 각 샘플에 대해 두 개의 확률 (0 클래스와 1 클래스에 속할 확률)이 반환됩니다. 다중 클래스 분류에서는 각 클래스에 대한 확률이 반환됩니다.