From 52d9dbf5af88ff93efac292bd5b3ecc964e6581a Mon Sep 17 00:00:00 2001 From: Spiros Maggioros Date: Wed, 6 Nov 2024 21:25:25 +0200 Subject: [PATCH] Reduced deprecation warnings on mlp --- spare_scores/mlp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spare_scores/mlp.py b/spare_scores/mlp.py index 27a6b0a..626e39c 100644 --- a/spare_scores/mlp.py +++ b/spare_scores/mlp.py @@ -254,7 +254,7 @@ def get_stats(self, y: np.ndarray, y_hat: np.ndarray) -> None: else: self.stats["MAE"].append(metrics.mean_absolute_error(y, y_hat)) self.stats["RMSE"].append( - metrics.mean_squared_error(y, y_hat, squared=False) + metrics.root_mean_squared_error(y, y_hat) ) self.stats["R2"].append(metrics.r2_score(y, y_hat))