diff --git a/projet_imt.py b/projet_imt.py index f7ca18812b899aa73e0d3106f9b9ade33cae4585..8905d53779854a0cb5634e2d37fbaf4fb25108fc 100644 --- a/projet_imt.py +++ b/projet_imt.py @@ -28,13 +28,16 @@ def main(): # visualisation des clusters en 2D visualize_clusters(reduced_embeddings_2d, cluster_labels, title="Clusters visualisés en 2D") # metriques de clustering - silhouette_score = compute_silhouette_scores(reduced_embeddings, labels) + silhouette_score = compute_silhouette_scores(reduced_embeddings, cluster_labels) # afficher les resultats - messages_par_cluster= afficher_messages_par_cluster(df,labels) + messages_par_cluster= afficher_messages_par_cluster(df, cluster_labels) + # afficher les messages avec les topics + resultats= print_topics_per_cluster(df, cluster_labels): - return messages_par_cluster + + return resultats if __name__ == "__main__": - messages_par_cluster = main() - print(messages_par_cluster) \ No newline at end of file + resultats = main() + print(resultats) \ No newline at end of file