Avoid sparse term similarity matrix computation when nonzero_limit <= 0

piskvorky · May 26, 2020 · be3a0b8 · be3a0b8
1 parent 2368932
commit be3a0b8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gensim/similarities/termsim.py b/gensim/similarities/termsim.py
@@ -267,6 +267,9 @@ def progress_bar(iterable):
             row_buffer.append(column_number)
             data_buffer.append(1.0)
 
+            if nonzero_limit <= 0:
+                continue
+
             t1 = dictionary[t1_index]
             num_nonzero = column_nonzero[t1_index]
             num_rows = nonzero_limit - num_nonzero