이번에는 검증 데이터셋을 회귀선(직선)으로 표현해 보겠습니다.

코드 3-27 검증 데이터셋을 사용한 회귀선 표현

plt.scatter(X_test, y_test, color='gray')
plt.plot(X_test, y_pred, color='red', linewidth=2)
plt.show()

다음 그림은 검증 데이터셋을 사용하여 회귀선으로 표현한 출력 결과입니다.

▲ 그림 3-26 선형 회귀 예측 결과를 회귀선으로 표현

어떤가요? 출력 결과 그림을 보면 회귀선이 실제 데이터와 비슷하다는 것을 확인할 수 있습니다.