Skip to content
Snippets Groups Projects
Commit cb505192 authored by MAFTOUH Mohammed Amine's avatar MAFTOUH Mohammed Amine
Browse files

Edit fonctions_clustering.py

parent fcbefc2c
No related branches found
No related tags found
No related merge requests found
...@@ -9,15 +9,3 @@ def agglomerative_clustering(reduced_embeddings, n_clusters): ...@@ -9,15 +9,3 @@ def agglomerative_clustering(reduced_embeddings, n_clusters):
agg_clustering = AgglomerativeClustering(n_clusters=n_clusters) agg_clustering = AgglomerativeClustering(n_clusters=n_clusters)
agg_labels = agg_clustering.fit_predict(reduced_embeddings) agg_labels = agg_clustering.fit_predict(reduced_embeddings)
return agg_labels 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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment