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

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

작성한 질문수

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

작업형2 모의문제3

9분 42초에 pred_proba

해결된 질문

작성

·

212

0

  • 9분 42초 쯤

print(roc_auc_score(y_val, pred_proba[:,1])))

pred_proba 다음에[:,1]을 붙이는 이유는 무엇인가요?

 

답변 1

0

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

이진 분류에서 predict_proba는

[0확률, 1확률]로 반환됩니다. [0.2, 0.8], [0.3, 0.7] 로 반환되었다고 보면 pred_proba[:,1]을 통해서 1확률(0.8, 0.7)만 뽑아내기 위함이에요

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

작성한 질문수

질문하기