get topics fn

c8259dff · De Silva K.C.C.C · 7b14a9fc · c8259dff
Commit c8259dff authored Nov 14, 2022 by De Silva K.C.C.C
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 6 deletions

Indexing/bert.py Indexing/bert.py +4 -6

No files found.
--- a/Indexing/bert.py
+++ b/Indexing/bert.py
 import nltk
 import question_generator as q_gen

-# nltk.download('words')
+

 from bertopic import BERTopic
 from nltk.corpus import words
@@ -18,12 +18,11 @@ def get_topics(file):
            docs.append(line.rstrip())

    topics, probabilities = model.fit_transform(docs)
-    #
+    
    print(model.get_topic_freq())
-    #
+    
    print('done')
-    #
-    # model.get_topic_freq().head(11)
+    

    print(model.get_topics())

@@ -39,5 +38,4 @@ def get_topics(file):

 def get_topics_new(text):
    topics = q_gen.get_keywords(text, q_gen.summarizer(text))
-    # print(topics)
    return topics