작성한 질문수
[퇴근후딴짓] 빅데이터 분석기사 실기 (작업형1,2,3)
작업형2 모의문제3
해결된 질문
작성
·
220
0
9분 42초 쯤
print(roc_auc_score(y_val, pred_proba[:,1])))
pred_proba 다음에[:,1]을 붙이는 이유는 무엇인가요?
답변 1
이진 분류에서 predict_proba는
[0확률, 1확률]로 반환됩니다. [0.2, 0.8], [0.3, 0.7] 로 반환되었다고 보면 pred_proba[:,1]을 통해서 1확률(0.8, 0.7)만 뽑아내기 위함이에요