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

몽상님의 프로필 이미지
몽상

작성한 질문수

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

체험 2유형 문제 질문

해결된 질문

작성

·

252

0

안녕하세요, 체험사이트 2유형 문제 풀어보다가 질문이 있습니다.

 

( https://dataq.goorm.io/exam/3/%EB%B9%85%EB%8D%B0%EC%9D%B4%ED%84%B0%EB%B6%84%EC%84%9D%EA%B8%B0%EC%82%AC-%EC%8B%A4%EA%B8%B0-%EC%B2%B4%ED%97%98/quiz/4 )

 

문제에 성별을 예측하는 모델을 만들고, 제출하는 csv 파일 내용 예시가 pred 0 0 0 1 이런 식으로 되어있는데

예측 결과는 ROC_AUC 평가지표로 평가한다고 되어있어서요.

이럴때는 predict를 써야할지, 아니면 predict_proba를 사용해서 pred[:,1]을 csv로 만들어야 할지요..

답변 1

1

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

문제가 명확하지 않더라고요

평가지표가 roc-auc면 predict_proba입니다!! 확률값을 제출해주세요

 

predict_proba를 사용해서 pred[:,1]을 csv로 만들어야 할지요.. 👌

몽상님의 프로필 이미지
몽상

작성한 질문수

질문하기