-
질문 & 답변
카테고리
-
세부 분야
딥러닝 · 머신러닝
-
해결 여부
미해결
사이킷런 input 2차원배열 관련 질문
21.08.02 11:13 작성 조회수 421
0
안녕하세요.
좋은 강의 해주셔서 감사합니다.
sklearn에 들어가는 자료의 배열에 대해 질문이 있습니다.
강의를 보면
train_input은 reshape을 해주는 반면,
train_target은 reshape을 해주지 않습니다.
사이킷런에서는 Input으로 들어가는 데이터에 대해서만 2차원 배열을 요구하나요?
그렇다면, target 데이터는 1차원이든 2차원이든 상관이 없나요?
답변을 작성해보세요.
0
박해선
지식공유자2021.08.02
안녕하세요. 박해선입니다. 일반적으로 사이킷런의 모델은 타깃으로 1차원 배열을 기대합니다. 만약 타깃을 (m, 1) 크기의 2차원 배열로 전달해도 (m,) 로 바꾸어 사용합니다. 다만 멀티아웃풋을 지원하는 사이킷런 모델은 (m, n_classes) 크기의 타깃을 사용할 수 있습니다. 멀티아웃풋을 지원하는 사이킷런 모델은 https://scikit-learn.org/stable/modules/multiclass.html 을 참고하세요. 감사합니다.
답변 1