Skip to content

Commit 7af6984

Browse files
Implement RMSE calculation in forecasting.py
Add function to calculate Root Mean Squared Error (RMSE)
1 parent 99b5dbd commit 7af6984

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

src/qmlhc/metrics/forecasting.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,3 +113,9 @@ def delta_lag(y_true_seq: Array, y_pred_seq: Array) -> float:
113113
sign_pred = np.sign(dt_pred)
114114
alignment = (sign_true * sign_pred).mean()
115115
return float(alignment)
116+
117+
def rmse(y_true: np.ndarray, y_pred: np.ndarray) -> float:
118+
"""Root Mean Squared Error (RMSE)."""
119+
y_true = np.asarray(y_true, dtype=float)
120+
y_pred = np.asarray(y_pred, dtype=float)
121+
return float(np.sqrt(np.mean((y_pred - y_true) ** 2)))

0 commit comments

Comments
 (0)