Inflearn Community Q&A
predict_proba 와 predict 차이
Resolved
Written on
·
1.6K
0
해당 문제는 predict_proba 로 해야 하는 이유가 무엇인가요 ?
아래코드처럼 predict 으로 하면 안되는 이유가 있나요 ?
문제에서 둘을 어떻게 구분해서 써야하는건지 모르겠어요.
pred = model.predict(X_val)
roc_auc_score(y_val, pred)
pred = model.predict(X_test)
submit = pd.DataFrame(
{
'cust_id':cust_id,
'gender':pred
}
)python머신러닝빅데이터pandas빅데이터분석기사
Answer 1
2
roadmap
Instructor
predict 함수:
각 입력 샘플에 대한 예측된 클래스 레이블을 반환합니다.
즉, 이진 분류 문제의 경우, 결과는 보통 0 또는1로 반환
predict_proba 함수:
각 입력 샘플에 대해 클래스별 확률을 반환합니다.
반환된 값은 각 클래스에 속할 확률을 나타내며, 이진 분류에서는 보통 두 개의 확률로 구성됩니다: [클래스 0일 확률, 클래스 1일 확률].
predict_proba 는 언제 사용하나요?
1(양성) 확률 값을 묻는다 그럼 predict_proba
평가지표가 roc-auc다 그럼 predict_proba
분류의 나머지는 거의 다 predict이고
회귀는 전부 predict로 하면 됩니다.





