diff --git a/ignite/metrics/clustering/calinski_harabasz_score.py b/ignite/metrics/clustering/calinski_harabasz_score.py index ddea539d9bd..fe58ac46151 100644 --- a/ignite/metrics/clustering/calinski_harabasz_score.py +++ b/ignite/metrics/clustering/calinski_harabasz_score.py @@ -5,6 +5,8 @@ from ignite.metrics.clustering._base import _ClusteringMetricBase +__all__ = ["CalinskiHarabaszScore"] + def _calinski_harabasz_score(features: Tensor, labels: Tensor) -> float: from sklearn.metrics import calinski_harabasz_score diff --git a/ignite/metrics/clustering/davies_bouldin_score.py b/ignite/metrics/clustering/davies_bouldin_score.py index c6433e7264f..b34ec69f51a 100644 --- a/ignite/metrics/clustering/davies_bouldin_score.py +++ b/ignite/metrics/clustering/davies_bouldin_score.py @@ -5,6 +5,8 @@ from ignite.metrics.clustering._base import _ClusteringMetricBase +__all__ = ["DaviesBouldinScore"] + def _davies_bouldin_score(features: Tensor, labels: Tensor) -> float: from sklearn.metrics import davies_bouldin_score diff --git a/ignite/metrics/clustering/silhouette_score.py b/ignite/metrics/clustering/silhouette_score.py index 2f4e82cd2d1..d82d603e0ea 100644 --- a/ignite/metrics/clustering/silhouette_score.py +++ b/ignite/metrics/clustering/silhouette_score.py @@ -5,6 +5,8 @@ from ignite.metrics.clustering._base import _ClusteringMetricBase +__all__ = ["SilhouetteScore"] + def _get_silhouette_score(**kwargs: Any) -> Callable[[Tensor, Tensor], float]: from sklearn.metrics import silhouette_score