PR piskvorky#986 merged in wmd in keyedvectors.py

anmolgulati · Oct 31, 2016 · 0690597 · 0690597
1 parent 7a6c281
commit 0690597
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gensim/models/keyedvectors.py b/gensim/models/keyedvectors.py
@@ -164,6 +164,10 @@ def wmdistance(self, document1, document2):
         dictionary = Dictionary(documents=[document1, document2])
         vocab_len = len(dictionary)
 
+        if vocab_len == 1:
+            # Both documents are composed by a single unique token
+            return 0.0
+
         # Sets for faster look-up.
         docset1 = set(document1)
         docset2 = set(document2)