funasr/models/campplus/cluster_backend.py
@@ -7,10 +7,10 @@ import scipy import torch import sklearn import hdbscan import numpy as np from sklearn.cluster._kmeans import k_means from sklearn.cluster import HDBSCAN class SpectralCluster: @@ -125,7 +125,7 @@ n_components=min(self.n_components, X.shape[0] - 2), metric=self.metric, ).fit_transform(X) labels = hdbscan.HDBSCAN( labels = HDBSCAN( min_samples=self.min_samples, min_cluster_size=self.min_cluster_size, allow_single_cluster=True).fit_predict(umap_X)