인프런 커뮤니티 질문&답변
rf.predict와 rf.predict_proba 차이
해결된 질문
작성
·
506
답변 1
0
퇴근후딴짓
지식공유자
문제에서 확률을 묻거나 평가지표가 roc-auc인 경우에는 predict_proba를 사용합니다. 
참고!
predict와 predict_proba는 분류 모델에서 사용되는 두 가지 다른 유형의 함수입니다. 이들의 주된 차이점은 반환하는 결과의 형태에 있습니다.
1. predict 함수
- 사용: - predict함수는 각 데이터 포인트에 대한 예측된 클래스 레이블을 반환합니다.
- 결과: 이진 분류 문제에서 결과는 0 또는 1과 같은 레이블입니다. 다중 클래스 분류에서는 각 샘플에 대해 하나의 클래스 레이블을 반환합니다. 
2. predict_proba 함수
- 사용: - predict_proba는 각 클래스에 대한 예측 확률을 반환합니다.
- 결과: 이진 분류에서는 각 샘플에 대해 두 개의 확률 (0 클래스와 1 클래스에 속할 확률)이 반환됩니다. 다중 클래스 분류에서는 각 클래스에 대한 확률이 반환됩니다. 






