diff --git a/clustering/fonctions_clustering.py b/clustering/agg_clustering.py similarity index 50% rename from clustering/fonctions_clustering.py rename to clustering/agg_clustering.py index 13052dda89c666ed3f178cb5a97dbb8364b7d1ee..b8a25e2de281bcc70e242b44dcf3e108f9d37890 100644 --- a/clustering/fonctions_clustering.py +++ b/clustering/agg_clustering.py @@ -9,15 +9,3 @@ def agglomerative_clustering(reduced_embeddings, n_clusters): agg_clustering = AgglomerativeClustering(n_clusters=n_clusters) agg_labels = agg_clustering.fit_predict(reduced_embeddings) return agg_labels - -def gaussian_mixture(reduced_embeddings, n_clusters): - """ - Applique le modèle de mélange gaussien (GMM) avec un nombre fixe de clusters. - - :param reduced_embeddings: embeddings - :param n_clusters: Nombre de clusters - :return: Labels prédits - """ - gmm = GaussianMixture(n_components=n_clusters, random_state=42) - gmm_labels = gmm.fit_predict(reduced_embeddings) - return gmm_labels \ No newline at end of file