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

daeho4.kim님의 프로필 이미지
daeho4.kim

작성한 질문수

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

모델링 및 평가(분류)

roc_auc_score 코드에서 에러 발생해요..

해결된 질문

작성

·

368

0

캡처.JPG안녕하세요. 모델링 및 평가(분류) 강의에서 마지막 부분에 roc_auc_score 넣어주는 부분에서 자꾸 에러가 발생합니다.

에러는 아래와 같이 뜨는데요, 어떤 부분에서 잘못됐는지 알려주세요..

"IndexError: index 1 is out of bounds for axis 1 with size 1"

감사합니다.

답변 5

1

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

원인을 찾았습니다. 정상적인 것 같아 저도 한참 보았네요 ㅎㅎ 오타입니다 :)

이 코드가 잘못되었어요!!! 이렇게 될 경우 y_tr이 모두 0이 됩니다.

예측결과도 모두 0이구요 그래서 한개의 클래스만 나와 에러가 발생했어요

'>50K'로 수정히면 정상적으로 작동됩니다.

y = (y_train['income'] == '>50').astype(int)
daeho4.kim님의 프로필 이미지
daeho4.kim
질문자

아하.. 제가 오타를 냈군요;; 에러없이 넘어가서 못찾았었더 같아요^^;;

알려주셔서 너무 감사드립니다~!

1

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

validation은 어떻게 나누셨나요? validation 데이터에서 0만 예측되었나? 의심됩니다.

혹시 전체 코드를 제게 링크로 공유해주실 수 있을까요?

1

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

  1. 런타임 - 모두 실행으로도 동일 할까요? 종종 셀 단위 실행이 잘못되어 데이터가 꼬이기도 합니다.

  2. 그래도 에러가 난다면 roc_auc_score 괄호 안에 있는 y_val과 pred[:,1] 데이터를 각각 출력해 줄 수 있을까요? 둘 중 하나가 뭔가 잘못된 것 같아요

0

daeho4.kim님의 프로필 이미지
daeho4.kim
질문자

선생님, 항상 빠른 답변 감사드립니다.

https://colab.research.google.com/drive/1QFM1eMv1WYYhSGEi41nJA4TxDdjkRsY8?usp=sharing

링크 공유 드립니다. 혹시 안열리시면 말씀해주세요~!

0

daeho4.kim님의 프로필 이미지
daeho4.kim
질문자

y_val 출력은 위와 같습니다.

근데 pred[:,1] 출력을 하면 동일한 에러가 뜹니다.

daeho4.kim님의 프로필 이미지
daeho4.kim

작성한 질문수

질문하기