인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

Inflearn Community Q&A

김성광's profile image
김성광

asked

[After-work activities] Big Data Analysis Engineer Practical (Work-type 1,2,3)

5th Exam, Past Exam Type (Type 2 - Hands-on)

rf.predict와 rf.predict_proba 차이

Resolved

Written on

·

460

0

  • rf.predict와 rf.predict_proba 차이

어떨때는 predict만 쓰시고 어떨때는 predict_proba를 쓰시던데

어떤 차이가 있고 뭘 써야하나요?

python머신러닝빅데이터pandas빅데이터분석기사

Answer 1

0

roadmap님의 프로필 이미지
roadmap
Instructor

문제에서 확률을 묻거나 평가지표가 roc-auc인 경우에는 predict_proba를 사용합니다.

참고!

predictpredict_proba는 분류 모델에서 사용되는 두 가지 다른 유형의 함수입니다. 이들의 주된 차이점은 반환하는 결과의 형태에 있습니다.

1. predict 함수

  • 사용: predict 함수는 각 데이터 포인트에 대한 예측된 클래스 레이블을 반환합니다.

  • 결과: 이진 분류 문제에서 결과는 0 또는 1과 같은 레이블입니다. 다중 클래스 분류에서는 각 샘플에 대해 하나의 클래스 레이블을 반환합니다.

     

2. predict_proba 함수

  • 사용: predict_proba는 각 클래스에 대한 예측 확률을 반환합니다.

  • 결과: 이진 분류에서는 각 샘플에 대해 두 개의 확률 (0 클래스와 1 클래스에 속할 확률)이 반환됩니다. 다중 클래스 분류에서는 각 클래스에 대한 확률이 반환됩니다.

김성광's profile image
김성광

asked

Ask a question