ffb9bab9595bebf625c01d58a8be77932e00a8f6,gensim/models/word2vec.py,Word2Vec,__getitem__,#Word2Vec#Any#,1119

Before Change

                    mean.append(self.syn0norm[self.vocab[w].index])
                else:
                    raise KeyError("word "%s" not in vocabulary" % w)
            mean = matutils.unitvec(array(mean).mean(axis=0)).astype(REAL)
            return mean
        else:
            return self.syn0[self.vocab[word].index]

After Change

          array([ -1.40128313e-02, ...]

        
        return array([self.syn0[self.vocab[word].index] for word in words])


    def __contains__(self, word):
        return word in self.vocab

Instances