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

hobum5263님의 프로필 이미지

작성한 질문수

혼자 공부하는 머신러닝+딥러닝

k-최근접 이웃 회귀

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

작성

·

516

0

안녕하세요. 

좋은 강의 해주셔서 감사합니다.

sklearn에 들어가는 자료의 배열에 대해 질문이 있습니다.

강의를 보면

train_input은 reshape을 해주는 반면, 

train_target은 reshape을 해주지 않습니다.

사이킷런에서는 Input으로 들어가는 데이터에 대해서만 2차원 배열을 요구하나요?

그렇다면, target 데이터는 1차원이든 2차원이든 상관이 없나요?

답변 1

0

박해선님의 프로필 이미지
박해선
지식공유자

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

hobum5263님의 프로필 이미지
hobum5263
질문자

상세한 답변 감사합니다. ㅎㅎ