강의

멘토링

커뮤니티

Inflearn Community Q&A

coolkim11255667's profile image
coolkim11255667

asked

[After Work Side Projects] Big Data Analytics Engineer Practical Exam (Task Types 1, 2, 3)

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님의 프로필 이미지
roadmap
Instructor

predict 함수:

  • 각 입력 샘플에 대한 예측된 클래스 레이블을 반환합니다.

  • 즉, 이진 분류 문제의 경우, 결과는 보통 0 또는1로 반환

 

predict_proba 함수:

  • 각 입력 샘플에 대해 클래스별 확률을 반환합니다.

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

 

predict_proba 는 언제 사용하나요?

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

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

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

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

coolkim11255667's profile image
coolkim11255667

asked

Ask a question