Convert ensembla LDA to LDA.

NLeSC · Feb 12, 2024 · b9ec0cb · b9ec0cb
1 parent 13634da
commit b9ec0cb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/litstudy/nlp.py b/litstudy/nlp.py
@@ -400,6 +400,8 @@ def train_elda_model(
     else:
         model = EnsembleLda.load(filename)
 
+    model = model.generate_gensim_representation()
+
     doc2topic = corpus2dense(model[freqs], num_topics).T
     topic2token = model.get_topics()