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

pred를 예측할 때 혹시 predict_proba를 사용했나요?

그렇다면 두번째인 양성(1) 컬럼을 선택해야 합니다.

pred[:,1]

안녕하세요, 인프런 AI 인턴입니다.

khu0605님께서 문의하신 ValueError: "Per-column arrays must each be 1-dimensional" 오류는 주로 데이터 프레임으로 변환 시 2차원 배열을 사용하려 할 때 발생합니다. 이 오류는 각 열에 들어가는 데이터가 1차원 배열이어야 하는데, 2차원 배열이나 일관되지 않는 형식을 사용했을 때 발생하는 경향이 있습니다.

다음 단계에 따라 문제를 해결해 보세요:

1. pred 변수의 차원 확인: pred 배열이 2차원일 가능성이 있으니 print(pred.shape)를 활용하여 해당 배열의 차원을 확인해 보세요. 만약 pred가 다차원 배열이라면, 이를 1차원으로 변환해야 합니다.

2. reshape 또는 flatten 사용: pred가 단일 예측 결과를 가진 2차원 배열이라면, .flatten() 메서드를 사용하여 1차원 배열로 변환할 수 있습니다.

   pred = pred.flatten()
   

   또한, .reshape(-1)을 사용하면 자동으로 1차원 배열로 바꿀 수 있습니다.

3. 정확한 데이터 확인: 항상 test.index와 pred가 같은 길이인지 확인하는 것이 중요합니다. 길이가 다를 경우에는 맞추어 주어야 오류 없이 저장할 수 있습니다.

비슷한 사례로 기존의 답변을 참고하시려면, 아래 링크를 확인해 보세요:

• 모의고사 풀어보기2 csv파일 내보내기 중 - CSV 파일 저장 시 발생할 수 있는 유사 오류에 대한 설명을 포함하고 있습니다.
  

저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다.
현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏
추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.