본문 바로가기
AI

[AI] 평균 제곱 오차(Mean Squared Error) 손실 함수

by Nhahan 2024. 9. 9.

 

f(x_i): 모델의 예측 값

y(i): 실제 값


 

현재 epoch의 손실값을 구하는 함수이다.

i = 1부터 N까지 모델의 예측 값에서 실제 값을 뺀 값을 제곱한 값을 모두 더한 뒤, N으로 나눠준다.

 

모델의 예측 값에서 실제 값을 뺀 값을 제곱하는 이유는 음수일수도 있기 때문이다.

 

epoch 회수 m회에 대하여 예측값을 모두 더하고 m으로 나눠주면 평균 손실값이 나오는데, 이것이 현재 모델의 성능이다.

 

 

 

 

 


p.s.

손실 함수에는 여러 종류가 있다.

 

 

댓글