shixian.shi
2024-01-23 ae4dceecf03c643be770e3274a070078e45fc9c7
funasr/models/campplus/cluster_backend.py
@@ -119,6 +119,7 @@
        self.metric = metric
    def __call__(self, X):
        from umap.umap_ import UMAP
        umap_X = umap.UMAP(
            n_neighbors=self.n_neighbors,
            min_dist=0.0,
@@ -156,6 +157,7 @@
        if X.shape[0] < 20:
            return np.zeros(X.shape[0], dtype='int')
        if X.shape[0] < 2048 or k is not None:
            # unexpected corner case
            labels = self.spectral_cluster(X, k)
        else:
            labels = self.umap_hdbscan_cluster(X)