MAINT Parameter Validation for feature_selection.f_regression (scikit-learn#25736)

ZeeshanLone · jeremiedbb · jpangas · commit 93acdf2c95c7 · 2023-03-06T23:01:50.000+03:00
Co-authored-by: Jérémie du Boisberranger &lt;34657725+jeremiedbb@users.noreply.github.com&gt;
diff --git a/sklearn/feature_selection/_univariate_selection.py b/sklearn/feature_selection/_univariate_selection.py
@@ -334,6 +334,14 @@ def r_regression(X, y, *, center=True, force_finite=True):
     return correlation_coefficient
 
 
+@validate_params(
+    {
+        "X": ["array-like", "sparse matrix"],
+        "y": ["array-like"],
+        "center": ["boolean"],
+        "force_finite": ["boolean"],
+    }
+)
 def f_regression(X, y, *, center=True, force_finite=True):
     """Univariate linear regression tests returning F-statistic and p-values.
 
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -115,6 +115,7 @@ def _check_function_param_validation(
     "sklearn.feature_extraction.image.reconstruct_from_patches_2d",
     "sklearn.feature_selection.chi2",
     "sklearn.feature_selection.f_classif",
+    "sklearn.feature_selection.f_regression",
     "sklearn.metrics.accuracy_score",
     "sklearn.metrics.auc",
     "sklearn.metrics.average_precision_score",
