speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #1596 from wanghuii1/main

Browse Source 
Update cluster_backend.py
This commit is contained in:
Shi Xian 2024-04-09 10:41:11 +08:00 committed by GitHub
commit c6574bf4f4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/models/campplus/cluster_backend.py
View File

@ -10,6 +10,7 @@ import sklearn
import numpy as np import numpy as np
from sklearn.cluster._kmeans import k_means from sklearn.cluster._kmeans import k_means
from sklearn.cluster import HDBSCAN
class SpectralCluster: class SpectralCluster:
@ -115,8 +116,6 @@ class UmapHdbscan:
self.min_samples = min_samples self.min_samples = min_samples
self.min_cluster_size = min_cluster_size self.min_cluster_size = min_cluster_size
self.metric = metric self.metric = metric
import hdbscan
self.hdbscan = hdbscan
def __call__(self, X): def __call__(self, X):
import umap.umap_ as umap import umap.umap_ as umap
@ -126,7 +125,7 @@ class UmapHdbscan:
n_components=min(self.n_components, X.shape[0] - 2), n_components=min(self.n_components, X.shape[0] - 2),
metric=self.metric, metric=self.metric,
).fit_transform(X) ).fit_transform(X)
labels = self.hdbscan.HDBSCAN( labels = HDBSCAN(
min_samples=self.min_samples, min_samples=self.min_samples,
min_cluster_size=self.min_cluster_size, min_cluster_size=self.min_cluster_size,
allow_single_cluster=True).fit_predict(umap_X) allow_single_cluster=True).fit_predict(umap_X)