Reduced deprecation warnings on mlp

CBICA · Nov 6, 2024 · 52d9dbf · 52d9dbf
1 parent 25e3efa
commit 52d9dbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))