MAINT Parameters validation for sklearn.metrics.mean_poisson_deviance (#26104)

ROMEEZHOU · web-flow · commit 500636b0cde2 · 2023-04-07T16:00:51.000+02:00
diff --git a/sklearn/metrics/_regression.py b/sklearn/metrics/_regression.py
@@ -1172,6 +1172,13 @@ def mean_tweedie_deviance(y_true, y_pred, *, sample_weight=None, power=0):
     )
 
 
+@validate_params(
+    {
+        "y_true": ["array-like"],
+        "y_pred": ["array-like"],
+        "sample_weight": ["array-like", None],
+    }
+)
 def mean_poisson_deviance(y_true, y_pred, *, sample_weight=None):
     """Mean Poisson deviance regression loss.
 
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -199,6 +199,7 @@ def _check_function_param_validation(
     "sklearn.metrics.mean_absolute_percentage_error",
     "sklearn.metrics.mean_gamma_deviance",
     "sklearn.metrics.mean_pinball_loss",
+    "sklearn.metrics.mean_poisson_deviance",
     "sklearn.metrics.mean_squared_error",
     "sklearn.metrics.mean_squared_log_error",
     "sklearn.metrics.mean_tweedie_deviance",
