MAINT Parameters validation for sklearn.metrics.log_loss (#25577)

JanFidor · web-flow · commit 6f6b4bdd9c94 · 2023-02-12T18:25:36.000+01:00
diff --git a/sklearn/metrics/_classification.py b/sklearn/metrics/_classification.py
@@ -41,7 +41,7 @@
 from ..utils.multiclass import type_of_target
 from ..utils.validation import _num_samples
 from ..utils.sparsefuncs import count_nonzero
-from ..utils._param_validation import StrOptions, Options, validate_params
+from ..utils._param_validation import StrOptions, Options, Interval, validate_params
 from ..exceptions import UndefinedMetricWarning
 
 from ._base import _check_pos_label_consistency
@@ -2569,6 +2569,16 @@ def hamming_loss(y_true, y_pred, *, sample_weight=None):
         raise ValueError("{0} is not supported".format(y_type))
 
 
+@validate_params(
+    {
+        "y_true": ["array-like"],
+        "y_pred": ["array-like"],
+        "eps": [StrOptions({"auto"}), Interval(Real, 0, 1, closed="both")],
+        "normalize": ["boolean"],
+        "sample_weight": ["array-like", None],
+        "labels": ["array-like", None],
+    }
+)
 def log_loss(
     y_true, y_pred, *, eps="auto", normalize=True, sample_weight=None, labels=None
 ):
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -122,6 +122,7 @@ def _check_function_param_validation(
     "sklearn.metrics.det_curve",
     "sklearn.metrics.f1_score",
     "sklearn.metrics.hamming_loss",
+    "sklearn.metrics.log_loss",
     "sklearn.metrics.mean_absolute_error",
     "sklearn.metrics.mean_squared_error",
     "sklearn.metrics.mean_tweedie_deviance",
