From 329bb09af43d920ad5d5b4e9cba6237c742fcc1b Mon Sep 17 00:00:00 2001 From: MAFTOUH Mohammed Amine <mohammed-amine.maftouh@imt-atlantique.net> Date: Fri, 28 Mar 2025 13:24:56 +0000 Subject: [PATCH] Add new file --- topic generation/print_topics_per_cluster.py | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 topic generation/print_topics_per_cluster.py diff --git a/topic generation/print_topics_per_cluster.py b/topic generation/print_topics_per_cluster.py new file mode 100644 index 0000000..3b248c9 --- /dev/null +++ b/topic generation/print_topics_per_cluster.py @@ -0,0 +1,10 @@ +def print_topics_per_cluster(df, agg_labels): + # Ajouter agg_labels comme colonne 'cluster' dans df + df['cluster'] = agg_labels + + print("\nTopics par cluster pour l'ensemble des données :") + for cluster in sorted(df['cluster'].unique()): + print(f"\nCluster {cluster}:") + messages_cluster = df[df['cluster'] == cluster]['message'].tolist() + topics = generate_topics_for_cluster(messages_cluster) + print("Topics générés:", topics) \ No newline at end of file -- GitLab