Added top words to the full report

Christoph · Nov 13, 2020 · 2199701 · 2199701
1 parent 9c30b60
commit 2199701
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/robics/robust_topics.py b/robics/robust_topics.py
@@ -674,6 +674,12 @@ def _compute_topic_stability(self):
                     "max": jensen_similarity.max(axis=1),
                 }
 
+                top_terms = {}
+                for i, t in enumerate(terms):
+                    top_terms['instance_'+str(i)] = t
+
+                report_full["top_words"] = top_terms
+
                 model.report.append(report)
                 model.report_full.append(report_full)
             # print("")