Merge pull request #17 from log0ymxm/patch-1

Fix assumption that gensim dictionary and corpus are built from same …
bmabey · Jul 24, 2015 · 69719cd · 69719cd
2 parents 6e7144b + 698e1c9
commit 69719cd
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pyLDAvis/gensim.py b/pyLDAvis/gensim.py
@@ -18,10 +18,9 @@ def _extract_data(topic_model, corpus, dictionary):
    doc_lengths = [sum([t[1] for t in doc]) for doc in corpus]
 
    term_freqs_dict = fp.merge_with(sum, *corpus)
-   N = len(term_freqs_dict)
 
-   vocab = [dictionary[id] for id in xrange(N)]
-   term_freqs = [term_freqs_dict[id] for id in xrange(N)]
+   vocab = [dictionary[id] for id in term_freqs_dict.keys()]
+   term_freqs = term_freqs_dict.values()
 
    gamma, _ = topic_model.inference(corpus)
    doc_topic_dists = _normalize(gamma)