BUG: Fix custom scorer rmsle

autogluon · Nov 22, 2024 · eb65b16 · eb65b16
1 parent 84fa2f9
commit eb65b16
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/autogluon/assistant/predictor.py b/src/autogluon/assistant/predictor.py
@@ -17,13 +17,14 @@
 
 
 def rmsle_func(y_true, y_pred, **kwargs):
+    y_pred = np.clip(y_pred, a_min=0.0, a_max=None)
     return np.sqrt(mean_squared_log_error(y_true, y_pred, **kwargs))
 
 
 root_mean_square_logarithmic_error = make_scorer(
     "root_mean_square_logarithmic_error",
     rmsle_func,
-    optimum=1,
+    optimum=0,
     greater_is_better=False,
 )