From cb5051923ab7114fa4b6fef87231f3845473b74a Mon Sep 17 00:00:00 2001
From: MAFTOUH Mohammed Amine <mohammed-amine.maftouh@imt-atlantique.net>
Date: Mon, 3 Mar 2025 13:36:29 +0000
Subject: [PATCH] Edit fonctions_clustering.py
---
.../{fonctions_clustering.py => agg_clustering.py} | 12 ------------
1 file changed, 12 deletions(-)
rename clustering/{fonctions_clustering.py => agg_clustering.py} (50%)
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 13052dd..b8a25e2 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
--
GitLab