From 68bcd9e5e60df4ef439221e982500c79f9231ce6 Mon Sep 17 00:00:00 2001 From: jiawei-zhang-a Date: Sun, 9 Apr 2023 20:44:50 -0400 Subject: [PATCH 1/2] calinski_harabasz_score --- sklearn/metrics/cluster/_unsupervised.py | 6 ++++++ 1 file changed, 6 insertions(+) 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. From 47578350abd9ed9559ff5d0d4fc1a47af69901b6 Mon Sep 17 00:00:00 2001 From: jiawei-zhang-a Date: Sun, 9 Apr 2023 20:44:59 -0400 Subject: [PATCH 2/2] calinski_harabasz_score --- sklearn/tests/test_public_functions.py | 1 + 1 file changed, 1 insertion(+) 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",