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. Calcule les scores de silhouette pour différents nombres de clusters.
:param reduced_embeddings: Matrice des embeddings réduits :param reduced_embeddings: Matrice des embeddings réduits
:param clustering_results: Dictionnaire contenant les labels prédits pour chaque nombre de clusters :param labels: les labels prédits par les algos de clustering
:return: Dictionnaire des scores de silhouette :return: silhouette score
""" """
silhouette_scores = {} silhouette_avg = silhouette_score(reduced_embeddings, labels)
for n_clusters, labels in clustering_results.items(): return silhouette_avg
silhouette_avg = silhouette_score(reduced_embeddings, labels) \ No newline at end of file
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