Skip to content

Commit f4a2d00

Browse files
committed
Fix the example code for SoftCosineSimilarity
1 parent 01c5336 commit f4a2d00

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

gensim/similarities/docsim.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -866,13 +866,13 @@ class SoftCosineSimilarity(interfaces.SimilarityABC):
866866
>>> from gensim.test.utils import common_texts
867867
>>> from gensim.corpora import Dictionary
868868
>>> from gensim.models import Word2Vec, WordEmbeddingSimilarityIndex
869-
>>> from gensim.similarities import SoftCosineSimilarity, TermSimilarityMatrix
869+
>>> from gensim.similarities import SoftCosineSimilarity, SparseTermSimilarityMatrix
870870
>>>
871871
>>> model = Word2Vec(common_texts, size=20, min_count=1) # train word-vectors
872-
>>> termsim_index = WordEmbeddingSimilarityIndex(model)
872+
>>> termsim_index = WordEmbeddingSimilarityIndex(model.wv)
873873
>>> dictionary = Dictionary(common_texts)
874874
>>> bow_corpus = [dictionary.doc2bow(document) for document in common_texts]
875-
>>> similarity_matrix = TermSimilarityMatrix(termsim_index, dictionary) # construct similarity matrix
875+
>>> similarity_matrix = SparseTermSimilarityMatrix(termsim_index, dictionary) # construct similarity matrix
876876
>>> docsim_index = SoftCosineSimilarity(bow_corpus, similarity_matrix, num_best=10)
877877
>>>
878878
>>> query = 'graph trees computer'.split() # make a query

0 commit comments

Comments
 (0)