인프런 커뮤니티 질문&답변

네 맞습니다. 제 스타일로 하다보니 그렇게 강의에서 사용했어요! 특별한 이유는 없습니다.

편한 방법으로 사용해주세요!!

회귀 평가지표 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}')