Pass num_topics to LdaModel

Pass the num_topics parameter to LdaModel with gensim4.
NLeSC · Jan 5, 2024 · db995c1 · db995c1
1 parent b739626
commit db995c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litstudy/nlp.py b/litstudy/nlp.py
@@ -326,7 +326,7 @@ def train_lda_model(corpus: Corpus, num_topics, seed=0, **kwargs) -> TopicModel:
         model = LdaModel(list(corpus), **kwargs)
     elif gensim_mayor == 4:
         from gensim.models.ldamodel import LdaModel
-        model = LdaModel(freqs, id2word=dic, **kwargs)
+        model = LdaModel(freqs, id2word=dic, num_topics=num_topics, **kwargs)
     else:
         from sys import exit
         exit('LdaModel could not be imported from gensim 3 or 4.')