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

김성광님의 프로필 이미지
김성광

작성한 질문수

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

5회 기출유형 (작업형2)

rf.predict와 rf.predict_proba 차이

해결된 질문

작성

·

440

0

  • rf.predict와 rf.predict_proba 차이

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

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

답변 1

0

퇴근후딴짓님의 프로필 이미지
퇴근후딴짓
지식공유자

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

참고!

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

1. predict 함수

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

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

     

2. predict_proba 함수

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

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

김성광님의 프로필 이미지
김성광

작성한 질문수

질문하기