단순 선형 회귀

 

다중 선형 회귀

코드의 형태는 크게 다르지 않습니다. 다만 x가 두 개가 되었으므로 x₁, x₂ 두 변수를 만들고, 기울기도 a₁과 a₂ 이렇게 두 개를 만듭니다. 앞서 했던 방법대로 경사 하강법을 적용해 보겠습니다. 먼저 예측 값을 구하는 식을 다음과 같이 세웁니다.

y_pred = a1 * x1 + a2 * x2 + b # 기울기와 절편 자리에 a1, a2, b를 각각 대입합니다.
error = y - y_pred           # 실제 값과 비교한 오차를 error로 놓습니다.