다음은 경사하강법을 이용해 W와 b를 구하는 간단한 예다(완벽한 선형 회귀 코드는 아니므로 데이터가 많아지면 오버플로(overflow) 에러가 발생할 수 있다).
# 연산을 위한 numpy와 차트를 그리기 위한 matplotlib를 임포트한다
import numpy as np
import matplotlib.pyplot as plt
# 회귀분석할 데이터를 준비한다
x_data = np.array( [ 1., 2., 3., 4., 5., 6. ] )
y_data = np.array( [ 9., 12., 15., 18., 21., 24. ] )
# 회귀분석에 앞서 matplotlib를 이용해 x_data와 y_data의 산포도를 그려본다
plt.figure( figsize=( 8,5 ) ) # 그림의 크기
plt.scatter( x_data, y_data ) # 산포도를 만든다
plt.show( ) # 산포도를 화면에 그린다
결과
▲ 그림 7-10 x_data와 y_data의 산포도