Make WordEmbeddingSimilarityIndex.most_similar easier to read

piskvorky · Jan 12, 2019 · aa08ad8 · aa08ad8
1 parent 290ae3a
commit aa08ad8
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/gensim/models/keyedvectors.py b/gensim/models/keyedvectors.py
@@ -1367,9 +1367,8 @@ def most_similar(self, t1, topn=10):
         if t1 not in self.keyedvectors.vocab:
             logger.debug('an out-of-dictionary term "%s"', t1)
         else:
-            for _, (t2, similarity) in zip(
-                    range(topn), self.keyedvectors.most_similar(
-                        positive=[t1], topn=topn, **self.kwargs)):
+            most_similar = self.keyedvectors.most_similar(positive=[t1], topn=topn, **self.kwargs)
+            for t2, similarity in most_similar:
                 if similarity > self.threshold:
                     yield (t2, similarity**self.exponent)