본문 바로가기
인공지능/Machine Learning

[Machine Learning] LinearRegression 성능측정 오차(error) 구하기

by coding_su 2022. 12. 1.

📝머신러닝 리니어 리그레이션 인공지능 성능측정 오차 구하기

오차(error)의 정의 : 실제값 - 예측값 (똑똑한 인공지능이란, 오차가 적은 것)

 

오차를 구하고 오차를 제곱한 후에 평균을 구한 값을 mean squared error라고 한다 (줄여서 MSE라고 한다)

MSE로 성능평가해서 수치가 작을수록 좋은 인공지능이다

# 실제값 - 예측값 변수로 저장
error = y_test - y_pred

# 저장한 변수를 제곱하고 평균을 구한다
(error ** 2).mean()

plt.plot을 사용해 실제값과 예측값을 차트로 그리기

plt.plot(y_test.values)
plt.plot(y_pred)
plt.legend(['Real', 'Pred'])
plt.show()

댓글