File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments