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

ajout des metriques d'évaluation du clustering

parent 550b469d
No related branches found
No related tags found
No related merge requests found
def compute_silhouette_scores(reduced_embeddings, clustering_results):
"""
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
"""
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
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