From 89b7e1907e3522a92b9b30d112fc2234c050eaf6 Mon Sep 17 00:00:00 2001
From: MAFTOUH Mohammed Amine <mohammed-amine.maftouh@imt-atlantique.net>
Date: Mon, 3 Mar 2025 13:36:59 +0000
Subject: [PATCH] Add new file

---
 clustering/gmm_clustering.py | 11 +++++++++++
 1 file changed, 11 insertions(+)
 create mode 100644 clustering/gmm_clustering.py

diff --git a/clustering/gmm_clustering.py b/clustering/gmm_clustering.py
new file mode 100644
index 0000000..000ab97
--- /dev/null
+++ b/clustering/gmm_clustering.py
@@ -0,0 +1,11 @@
+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