MAINT Parameters validation for sklearn.metrics.davies_bouldin_score (s…

…cikit-learn#26136)
neurodata · Apr 12, 2023 · f89af36 · f89af36
1 parent e74041f
commit f89af36
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/sklearn/metrics/cluster/_unsupervised.py b/sklearn/metrics/cluster/_unsupervised.py
@@ -358,6 +358,12 @@ def calinski_harabasz_score(X, labels):
  )
 
 
+@validate_params(
+ {
+ "X": ["array-like"],
+ "labels": ["array-like"],
+ }
+)
 def davies_bouldin_score(X, labels):
  """Compute the Davies-Bouldin score.
 

diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -182,6 +182,7 @@ def _check_function_param_validation(
  "sklearn.metrics.d2_absolute_error_score",
  "sklearn.metrics.d2_pinball_score",
  "sklearn.metrics.d2_tweedie_score",
+ "sklearn.metrics.davies_bouldin_score",
  "sklearn.metrics.dcg_score",
  "sklearn.metrics.det_curve",
  "sklearn.metrics.explained_variance_score",