diff --git a/gensim/similarities/docsim.py b/gensim/similarities/docsim.py index ee73328ff1..cdb966547d 100644 --- a/gensim/similarities/docsim.py +++ b/gensim/similarities/docsim.py @@ -1015,7 +1015,7 @@ class WmdSimilarity(interfaces.SimilarityABC): >>> >>> model = Word2Vec(common_texts, vector_size=20, min_count=1) # train word-vectors >>> - >>> index = WmdSimilarity(common_texts, model) + >>> index = WmdSimilarity(common_texts, model.wv) >>> # Make query. >>> query = ['trees'] >>> sims = index[query] @@ -1096,7 +1096,7 @@ def get_similarities(self, query): return result def __str__(self): - return "%s<%i docs, %i features>" % (self.__class__.__name__, len(self), self.w2v_model.wv.syn0.shape[1]) + return "%s<%i docs, %i features>" % (self.__class__.__name__, len(self), self.wv.vectors.shape[1]) class SparseMatrixSimilarity(interfaces.SimilarityABC): diff --git a/gensim/test/test_similarities.py b/gensim/test/test_similarities.py index 35ddd03397..0b917980d2 100644 --- a/gensim/test/test_similarities.py +++ b/gensim/test/test_similarities.py @@ -373,6 +373,11 @@ def test_iter(self): self.assertTrue(numpy.alltrue(sims >= 0.0)) self.assertTrue(numpy.alltrue(sims <= 1.0)) + @unittest.skipIf(PYEMD_EXT is False, "pyemd not installed") + def test_str(self): + index = self.cls(TEXTS, self.w2v_model) + self.assertTrue(str(index)) + class TestSoftCosineSimilarity(_TestSimilarityABC): def setUp(self):