def most_similar(self, word, N=10):
w = self._vocab.index(word)
sims = self._vecs.dot(self._vecs[w])
sims = heapq.nlargest(N, zip(sims,self._vocab))
return sims
微信
公众号
扫码关注公众号