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

kd03100님의 프로필 이미지
kd03100

작성한 질문수

[개정판] 파이썬 머신러닝 완벽 가이드

다항회귀를 이용한 과소적합 및 과적합 이해

3차 다항식 만들기

작성

·

385

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개의 레코드를 가지도록 만들었습니다.

감사합니다.

kd03100님의 프로필 이미지
kd03100

작성한 질문수

질문하기