• 카테고리

    질문 & 답변
  • 세부 분야

    딥러닝 · 머신러닝

  • 해결 여부

    미해결

3차 다항식 만들기

23.03.15 12:10 작성 조회수 318

0

선생님! 다항회귀공부하면서 코드에 3차 다항식을 만들 때

y = 1 + 2*X[:,0] + 3*X[:,0]**2 + 4*X[:,1]**3

저기 [:,0] 이부분은 식에 왜 해주는 건가요??

답변 1

답변을 작성해보세요.

0

안녕하십니까,

두개의 레코드를 반환하기 위해서 X[: , 0] 과 같이 X array에 인덱싱을 [:, 0] 과 같이 적용했습니다.

X array는 2차원 array로

[[0 1] [2 3]]

가 됩니다. 즉 shape가 (2, 2) 두개의 row와 두개의 column을 가지며 feature 데이터 세트를 의미 합니다.

첫번째 column에 해당하는 두개의 레코드인 [0, 2]는 X[:, 0] 을 통해 가져 올 수 있으며 이는 1차원 array가 됩니다. X[:1, 1]은 두번째 column에 해당하는 [1, 3]을 가져 오게 되며 마찬가지로 1차원 array가 됩니다.

y는 결과식 값으로 1차원 데이터인데 2개의 레코드를 가지게 됩니다. feature가 2x2 shape이므로 2개의 row를 가지므로 y결과 값은 1차원으로 2개의 레코드를 가지도록 만들었습니다.

감사합니다.