From 619824b070f0515973366e3459a3381bdbc31e52 Mon Sep 17 00:00:00 2001
From: MAFTOUH Mohammed Amine <mohammed-amine.maftouh@imt-atlantique.net>
Date: Mon, 3 Mar 2025 12:44:46 +0000
Subject: [PATCH] Edit mesures_clustering.py

---
 clustering/mesures_clustering.py | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/clustering/mesures_clustering.py b/clustering/mesures_clustering.py
index ed08e4b..af1ffdf 100644
--- a/clustering/mesures_clustering.py
+++ b/clustering/mesures_clustering.py
@@ -1,14 +1,10 @@
-def compute_silhouette_scores(reduced_embeddings, clustering_results):
+def compute_silhouette_scores(reduced_embeddings, labels):
     """
     Calcule les scores de silhouette pour différents nombres de clusters.
     
     :param reduced_embeddings: Matrice des embeddings réduits
-    :param clustering_results: Dictionnaire contenant les labels prédits pour chaque nombre de clusters
-    :return: Dictionnaire des scores de silhouette
+    :param labels: les labels prédits par les algos de clustering
+    :return: silhouette score
     """
-    silhouette_scores = {}
-    for n_clusters, labels in clustering_results.items():
-        silhouette_avg = silhouette_score(reduced_embeddings, labels)
-        silhouette_scores[n_clusters] = silhouette_avg
-        print(f"Nombre de clusters: {n_clusters}, Silhouette Score: {silhouette_avg:.2f}")
-    return silhouette_scores
\ No newline at end of file
+    silhouette_avg = silhouette_score(reduced_embeddings, labels)
+    return silhouette_avg
\ No newline at end of file
-- 
GitLab