다음 그림은 회귀 모델에 대한 예측 실행 결과입니다.

▲ 그림 3-25 선형 회귀 예제 예측 결과

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

코드 3-27 테스트 데이터셋을 사용한 회귀선 표현

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