다음으로 회귀분석 결과와 실제 데이터 간의 관계를 차트로 나타내보자. 점은 실제 데이터 y_data이고, 선은 기울기와 절편으로 예측한 y_pred이다.
# 앞서 구한 기울기와 절편을 이용해 그래프 그리기
y_pred = W * x_data + b
plt.scatter( x_data, y_data )
plt.plot( [min(x_data), max(x_data)], [min(y_pred), max(y_pred)] )
plt.show( )
결과
▲ 그림 7-11 회귀분석의 결과