Make Soft Cosine Measure support non-contiguous dictionaries (cont)

Witiko · Jul 2, 2018 · 9f8d0e8 · 9f8d0e8
1 parent 227d09e
commit 9f8d0e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gensim/similarities/termsim.py b/gensim/similarities/termsim.py
@@ -82,7 +82,7 @@ def __init__(self, dictionary, term_similarity=0.5):
 
     def most_similar(self, t1, topn=10):
         for __, (t2_index, t2) in zip(range(topn), (
-                (t2_index, t2) for t2_index, t2 in self.dictionary.items() if t2 != t1)):
+                (t2_index, t2) for t2_index, t2 in sorted(self.dictionary.items()) if t2 != t1)):
             yield (t2, self.term_similarity)
 
 
@@ -161,7 +161,7 @@ def __init__(self, source, dictionary=None, tfidf=None, symmetric=True, nonzero_
 
         if tfidf is None:
             logger.info("iterating over columns in dictionary order")
-            columns = sorted(dictionary.keys)
+            columns = sorted(dictionary.keys())
         else:
             assert max(tfidf.idfs) == matrix_order - 1
             logger.info("iterating over columns in tf-idf order")