사이킷런 input 2차원배열 관련 질문

Question

안녕하세요. 좋은 강의 해주셔서 감사합니다. sklearn에 들어가는 자료의 배열에 대해 질문이 있습니다. 강의를 보면 train_input은 reshape을 해주는 반면, train_target은 reshape을 해주지 않습니다. 사이킷런에서는 Input으로 들어가는 데이터에 대해서만 2차원 배열을 요구하나요? 그렇다면, target 데이터는 1차원이든 2차원이든 상관이 없나요?

박해선 · Answer

안녕하세요. 박해선입니다. 일반적으로 사이킷런의 모델은 타깃으로 1차원 배열을 기대합니다. 만약 타깃을 (m, 1) 크기의 2차원 배열로 전달해도 (m,) 로 바꾸어 사용합니다. 다만 멀티아웃풋을 지원하는 사이킷런 모델은 (m, n_classes) 크기의 타깃을 사용할 수 있습니다. 멀티아웃풋을 지원하는 사이킷런 모델은 https://scikit-learn.org/stable/modules/multiclass.html 을 참고하세요. 감사합니다.