Merge pull request #281 from rssdev10/fix/showprogress

Fix/showprogress
JuliaText · Feb 7, 2024 · 3ec01c8 · 3ec01c8
2 parents e130003 + 2c35760
commit 3ec01c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/lda.jl b/src/lda.jl
@@ -69,7 +69,7 @@ function lda(dtm::DocumentTermMatrix, ntopics::Int, iteration::Int,
     wait_time = showprogress ? 1.0 : Inf
 
     # Gibbs sampling
-    @showprogress wait_time for _ in 1:iteration
+    @showprogress dt=wait_time for _ in 1:iteration
         for doc in docs
             for (i, word) in enumerate(doc.text)
                 topicid_current = doc.topic[i]
@@ -89,7 +89,7 @@ function lda(dtm::DocumentTermMatrix, ntopics::Int, iteration::Int,
                 # select new topic
                 select = rand()
                 sum_of_prob = 0.0
-                new_topicid = 0
+                new_topicid = 1
                 for (selected_topicid, prob) in enumerate(probs)
                     sum_of_prob += prob / normalize_probs
                     if select < sum_of_prob