인프런 커뮤니티 질문&답변
작성한 질문수
무슨 에러인가요?답변 1
1
퇴근후딴짓
지식공유자
predict로 예측하는 것과
predict_proba로 예측하는 것은 다릅니다
이진 분류일때 전자는 클래스 0또는 1을 예측값으로 출력하고
후자는 0확률과 1확률을 출력해요
이때 pred변수에는 컬럼이 2개가 되는거죠
그런데 csv만들때 gender:pred라고 적으면 컬럼명은 gender하나인데 pred는 컬럼이 두 개니 에러발생 했습니다
이진분류일 때 1확률(양성) 하나만 선택해주세요
pred[:,1]