Make SparseTermSimilarityMatrix constructor defensive about most_simi…

…lar(topn=0)
piskvorky · May 17, 2019 · b588bd7 · b588bd7
1 parent b34b06e
commit b588bd7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gensim/similarities/termsim.py b/gensim/similarities/termsim.py
@@ -231,8 +231,9 @@ def __init__(self, source, dictionary=None, tfidf=None, symmetric=True, positive
             num_rows = nonzero_limit - num_nonzero
             most_similar = [
                 (dictionary.token2id[term], similarity)
-                for term, similarity in index.most_similar(t1, num_rows)
-                if term in dictionary.token2id]
+                for term, similarity in index.most_similar(t1, topn=num_rows)
+                if term in dictionary.token2id
+            ] if num_rows > 0 else []
 
             if tfidf is None:
                 rows = sorted(most_similar)