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

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

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

이제 거의 다 왔습니다. 마지막으로 모델을 평가해 보겠습니다. 선형 회귀는 평균 제곱 오차(평균제곱법)와 루트 평균 제곱 오차(루트 평균제곱법)를 사용하여 모델을 평가합니다.

코드 3-28 선형 회귀 모델 평가

print('평균제곱법:', metrics.mean_squared_error(y_test, y_pred))
print('루트 평균제곱법:', np.sqrt(metrics.mean_squared_error(y_test, y_pred)))