From 04f73b32d4f30264f61fe50645611f68fe9cf9ed Mon Sep 17 00:00:00 2001 From: Hao <1097261154@qq.com> Date: Tue, 19 Mar 2024 16:09:46 +0800 Subject: [PATCH] change the dim to a safe n_components --- raptor/cluster_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/raptor/cluster_utils.py b/raptor/cluster_utils.py index bdd739c..9b87384 100644 --- a/raptor/cluster_utils.py +++ b/raptor/cluster_utils.py @@ -69,7 +69,7 @@ def GMM_cluster(embeddings: np.ndarray, threshold: float, random_state: int = 0) def perform_clustering( embeddings: np.ndarray, dim: int, threshold: float, verbose: bool = False ) -> List[np.ndarray]: - reduced_embeddings_global = global_cluster_embeddings(embeddings, dim) + reduced_embeddings_global = global_cluster_embeddings(embeddings, min(dim, len(embeddings) -2)) global_clusters, n_global_clusters = GMM_cluster( reduced_embeddings_global, threshold )