diff --git a/clustering/gmm_clustering.py b/clustering/gmm_clustering.py new file mode 100644 index 0000000000000000000000000000000000000000..000ab979ef042ae2437040d332fc60a7daa7972b --- /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