[퇴근후딴짓] 빅데이터 분석기사 실기 (작업형1,2,3)

회귀 평가지표 RMSE를 구하는 3가지 방법입니다.

꼭 함수를 만들지 않아도 괜찮습니다.

mse를 구한다음 제곱하거나

squared=False 설정을 사용해 주세요 💪

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}')