4 파이썬 코딩으로 확인하는 다중 선형 회귀

지금까지 배운 내용을 토대로 다중 선형 회귀를 만들어 보겠습니다. 이번에는 x 값이 두 개이므로 다음과 같이 공부 시간 x₁, 과외 시간 x₂, 성적 y의 넘파이 배열을 만듭니다.

x1 = np.array([2, 4, 6, 8])
x2 = np.array([0, 4, 2, 3])
y = np.array([81, 93, 91, 97])

데이터의 분포를 그래프로 표현해 보면 다음과 같습니다.

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter3D(x1, x2, y);
plt.show()

그림 5-6 | 축이 하나 더 늘어 3D로 배치된 모습

앞서 x와 y 두 개의 축이던 것과는 달리 x₁, x₂, y 이렇게 세 개의 축이 필요합니다. 새로운 변수가 추가되면 차원이 하나씩 추가되면서 계산은 더욱 복잡해지는 것을 알 수 있습니다.