인프런 커뮤니티 질문&답변
작성한 질문수
RMSE 방법
해결된 질문
작성
·
118
답변 1
0
퇴근후딴짓
지식공유자
네 맞습니다. 제 스타일로 하다보니 그렇게 강의에서 사용했어요! 특별한 이유는 없습니다.
편한 방법으로 사용해주세요!!
회귀 평가지표 RMSE를 구하는 3가지 방법입니다.
꼭 함수를 만들지 않아도 괜찮습니다.
from sklearn.metrics import mean_squared_error
import numpy as np
# 예제 데이터
y_true = np.array([3, -0.5, 2, 7])
y_pred = np.array([2.5, 0.0, 2, 8])
# MSE 계산
mse = mean_squared_error(y_true, y_pred)
print(f'MSE: {mse}')
# RMSE 방법1 (squared=False)
rmse = mean_squared_error(y_true, y_pred, squared=False)
print(f'RMSE: {rmse}')
# RMSE 방법2
rmse = np.sqrt(mse)
print(f'RMSE(np): {rmse}')
# RMSE 방법3
rmse = mse ** 0.5
print(f'RMSE ** 0.5: {rmse}')