diff --git a/sklearn/metrics/cluster/_unsupervised.py b/sklearn/metrics/cluster/_unsupervised.py index fdb09d5d511fe..c181fdff4aaa9 100644 --- a/sklearn/metrics/cluster/_unsupervised.py +++ b/sklearn/metrics/cluster/_unsupervised.py @@ -298,6 +298,12 @@ def silhouette_samples(X, labels, *, metric="euclidean", **kwds): return np.nan_to_num(sil_samples) +@validate_params( + { + "X": ["array-like"], + "labels": ["array-like"], + } +) def calinski_harabasz_score(X, labels): """Compute the Calinski and Harabasz score. diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py index a4a9dbd9db739..bc96d4608e069 100644 --- a/sklearn/tests/test_public_functions.py +++ b/sklearn/tests/test_public_functions.py @@ -166,6 +166,7 @@ def _check_function_param_validation( "sklearn.metrics.average_precision_score", "sklearn.metrics.balanced_accuracy_score", "sklearn.metrics.brier_score_loss", + "sklearn.metrics.calinski_harabasz_score", "sklearn.metrics.class_likelihood_ratios", "sklearn.metrics.classification_report", "sklearn.metrics.cluster.adjusted_mutual_info_score",