Merge pull request #1389 from chinmayapancholi13/lda_update_logic_fix

Update logic to set 'updatetype' variable in LdaModel.py
piskvorky · Jun 5, 2017 · 3e3150c · 3e3150c
2 parents 214fff7 + 716ee4f
commit 3e3150c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gensim/models/ldamodel.py b/gensim/models/ldamodel.py
@@ -595,6 +595,10 @@ def update(self, corpus, chunksize=None, decay=None, offset=None,
 
         if update_every:
             updatetype = "online"
+            if passes == 1:
+                updatetype += " (single-pass)"
+            else:
+                updatetype += " (multi-pass)"
             updateafter = min(lencorpus, update_every * self.numworkers * chunksize)
         else:
             updatetype = "batch"