diff --git a/topic generation/print_topics_per_cluster.py b/topic generation/print_topics_per_cluster.py index 3b248c965d22873137ceeecd2213a8036806e06f..62ae9d4cc06890794e624577ecdd0f0bb473b9cb 100644 --- a/topic generation/print_topics_per_cluster.py +++ b/topic generation/print_topics_per_cluster.py @@ -1,6 +1,6 @@ -def print_topics_per_cluster(df, agg_labels): - # Ajouter agg_labels comme colonne 'cluster' dans df - df['cluster'] = agg_labels +def print_topics_per_cluster(df, labels): + # Ajouter labels comme colonne 'cluster' dans df + df['cluster'] = labels print("\nTopics par cluster pour l'ensemble des données :") for cluster in sorted(df['cluster'].unique()):