-
질문 & 답변
카테고리
-
세부 분야
자격증 (데이터 사이언스)
-
해결 여부
해결됨
랜포 모델로 파일을 제출하려 하는데 predict_proba를 사용하나요?
24.05.26 17:51 작성 조회수 90
0
5-2 type2 문제 (에어비앤비 가격) 문제에서는 pred = model.predict(test) 를 사용하는데5-3 type3 문제 (심장마비 확률) 문제에서 랜덤포레스트를 사용할 경우에도 pred=model.predict_proba(test)를 사용하나요? 평가지표에 roc-auc가 있으면 무조건 pred=model.predict_proba(test)로 예측한 후 DF변환을 해야 하나요? 아님 다른 기준이 있나요?
답변을 작성해보세요.
0
퇴근후딴짓
지식공유자2024.05.27
이진분류(0과 1)에서 평기지표가 roc-auc면 predict_proba로 예측하고 그 결과가 0확률과 1확률이 나옵니다.
그 중에 문제에서 1확률(양성)을 요구한다면 1확률만 넣어서 제출하면됩니다.
그외 평가지표는 predict만 사용하면됩니다.
그럼 roc-auc일 때 predict로 제출하면 틀린거냐?
그건 아니에요~ 수식으로 계산되기 때문에 틀리진 않아요. 다만 성능에 약간의 차이가 있을 뿐입니다.💪
답변 1