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

Edit mesures_clustering.py

parent 026c30e2
No related branches found
No related tags found
No related merge requests found
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
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