a31eab99dfbc6dcb7fe2ef149c59a30910adbbbd,src/gensim/similarities/docsim.py,SparseMatrixSimilarity,getSimilarities,#SparseMatrixSimilarity#Any#,206

Before Change

        // compute cosine similarity against every other document in the collection
        allSims = self.corpus * vec.tocsc() // N x T * T x 1 = N x 1
        allSims = list(allSims.toarray().flat) // convert to plain python list
        assert len(allSims) == self.corpus.shape[0] // make sure no document got lost!
        return allSims
//endclass SparseMatrixSimilarity

After Change

        // compute cosine similarity against every other document in the collection
        result = self.corpus * query.tocsc() // N x T * T x C = N x C
        if result.shape[1] == 1:
            // for queries of one document, return a 1d array
            result = result.toarray().flatten()
        else:
            // otherwise, return a 2d matrix (//queries x //index)
            result = result.toarray().T
        return result
//endclass SparseMatrixSimilarity

Instances