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

송승하님의 프로필 이미지

작성한 질문수

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

작업형2 모의문제3

랜포 모델로 파일을 제출하려 하는데 predict_proba를 사용하나요?

해결된 질문

24.05.26 17:51 작성

·

124

0

 5-2 type2 문제 (에어비앤비 가격) 문제에서는 pred = model.predict(test) 를 사용하는데5-3 type3 문제 (심장마비 확률) 문제에서 랜덤포레스트를 사용할 경우에도 pred=model.predict_proba(test)를 사용하나요? 평가지표에 roc-auc가 있으면 무조건 pred=model.predict_proba(test)로 예측한 후 DF변환을 해야 하나요? 아님 다른 기준이 있나요?

답변 1

0

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

2024. 05. 27. 09:59

이진분류(0과 1)에서 평기지표가 roc-aucpredict_proba로 예측하고 그 결과가 0확률과 1확률이 나옵니다.

그 중에 문제에서 1확률(양성)을 요구한다면 1확률만 넣어서 제출하면됩니다.

그외 평가지표는 predict만 사용하면됩니다.

 

그럼 roc-auc일 때 predict로 제출하면 틀린거냐?

그건 아니에요~ 수식으로 계산되기 때문에 틀리진 않아요. 다만 성능에 약간의 차이가 있을 뿐입니다.💪