diff --git a/sklearn/metrics/cluster/_supervised.py b/sklearn/metrics/cluster/_supervised.py index 6b8db4106151f..028e161634a82 100644 --- a/sklearn/metrics/cluster/_supervised.py +++ b/sklearn/metrics/cluster/_supervised.py @@ -1203,6 +1203,11 @@ def fowlkes_mallows_score(labels_true, labels_pred, *, sparse=False): return np.sqrt(tk / pk) * np.sqrt(tk / qk) if tk != 0.0 else 0.0 +@validate_params( + { + "labels": ["array-like"], + } +) def entropy(labels): """Calculate the entropy for a labeling. diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py index 62513b0f63cce..b28aa3f7c660e 100644 --- a/sklearn/tests/test_public_functions.py +++ b/sklearn/tests/test_public_functions.py @@ -173,6 +173,7 @@ def _check_function_param_validation( "sklearn.metrics.classification_report", "sklearn.metrics.cluster.adjusted_mutual_info_score", "sklearn.metrics.cluster.contingency_matrix", + "sklearn.metrics.cluster.entropy", "sklearn.metrics.cluster.fowlkes_mallows_score", "sklearn.metrics.cluster.homogeneity_completeness_v_measure", "sklearn.metrics.cluster.normalized_mutual_info_score",