From d13ab99cf0de37d79433b1b935dd389f7fecc2fb Mon Sep 17 00:00:00 2001 From: Maxim Ziatdinov Date: Sun, 27 Oct 2024 12:43:28 -0700 Subject: [PATCH] fix rmse util --- neurobayes/utils/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neurobayes/utils/utils.py b/neurobayes/utils/utils.py index a2dc449..57c16ed 100644 --- a/neurobayes/utils/utils.py +++ b/neurobayes/utils/utils.py @@ -92,8 +92,8 @@ def rmse(y_pred: jnp.ndarray, y_true: jnp.ndarray) -> jnp.ndarray: """ Calculates the root mean squared error between true and predicted values. """ - mse = mse(y_pred, y_true) - return jnp.sqrt(mse) + mse_ = mse(y_pred, y_true) + return jnp.sqrt(mse_) def mae(y_pred: jnp.ndarray, y_true: jnp.ndarray) -> jnp.ndarray: