如何获取numpy.ndarray中的键值对？ (Gensim Word2vec)

Question

我正在尝试获取密钥以及向量 model.syn0 中的向量，它通过 model.syn0["word"] 给出向量，它给出一个 n-dim 向量。有没有更好的方法来创建模型中所有单词的列表，其顺序与 syn0 的向量相同？我有 350000 个单词，这会花很长时间。

from gensim.models import word2vec as wv
model = wv.Word2Vec.load('model')
lab=[]
for i in model.syn0:
    lab.append(model.similar_by_vector(i)[0])

print(type(model.syn0))
    <type 'numpy.ndarray'>

Answer 1

在Gordon Mohr的方向，发现键值对存放在model.index2word中。

因此键值对列表可以很容易地通过以下方式获得：

lab=model.index2word

如何获取numpy.ndarray中的键值对？ (Gensim Word2vec)

How to get the key value pairs in numpy.ndarray? (Gensim Word2vec)

python

performance

gensim

word2vec