➋ 앞서 배운 경사 하강법을 실행하려면 옵티마이저에 sgd라고 설정합니다. 손실 함수는 평균 제곱 오차를 사용할 것이므로 mse라고 설정합니다. ➌ 끝으로 앞서 따로 적어 주었던 epochs 숫자를 model.fit() 함수에 적습니다.

학습 시간(x)이 입력되었을 때의 예측 점수는 model.predict(x)로 알 수 있습니다. 예측 점수로 그래프를 그려 보면 다음과 같습니다.

plt.scatter(x, y)
plt.plot(x, model.predict(x), 'r')    # 예측 결과를 그래프로 나타냅니다.
plt.show()

이제 모든 코드를 모아 보면 다음과 같습니다.