[Python] predict vs predict_proba 차이

솜솜·2023년 4월 27일
0

predict() 메소드는 범주를 예측하여 반환하고,
predict_praba() 메소드는 확률(probability)을 반환합니다.

## prediction for test set
y_pred = clf.predict(X_test) # class
y_pred_proba = clf.predict_proba(X_test) # probability

1) predict(test)
predict의 경우, 각각 모델의 최종 예측값 출력
ex) 회귀 모델 : 0.3456... 특정 값 출력
분류 모델 : 0,1,2,3 등 가장 확률 높은 클래스 출력

2) predict_proba(test)
분류 모델이 각 클래스에 대한 확률을 출력 (회귀에는 사용 불가)

profile
[Data Science] 차근차근 쌓아나가는

0개의 댓글