predict_proba 와 predict 차이

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

coolkim1125

작성한 질문수

[퇴근후딴짓] 빅데이터 분석기사 실기 (작업형1,2,3)

예시문제 작업형2(구 버전): 학습시간 최소화를 위해 구 버전은 삭제하고 있어요!

해결된 질문

작성

1.4K

해당 문제는 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 빅데이터분석기사

답변 1

퇴근후딴짓

지식공유자

predict 함수:

각 입력 샘플에 대한 예측된 클래스 레이블을 반환합니다.
즉, 이진 분류 문제의 경우, 결과는 보통 0 또는1로 반환

predict_proba 함수:

각 입력 샘플에 대해 클래스별 확률을 반환합니다.
반환된 값은 각 클래스에 속할 확률을 나타내며, 이진 분류에서는 보통 두 개의 확률로 구성됩니다: [클래스 0일 확률, 클래스 1일 확률].

predict_proba 는 언제 사용하나요?

1(양성) 확률 값을 묻는다 그럼 predict_proba

평가지표가 roc-auc다 그럼 predict_proba

분류의 나머지는 거의 다 predict이고

회귀는 전부 predict로 하면 됩니다.

coolkim1125

작성한 질문수

전체 Q&A

질문하기