如何在Word2Vec中检索数组

问题描述

我正在尝试在训练有素的word2vec模型中检索单词的数组/向量。在SpaCy中，可以通过model.vocab.get_vector（“ word”）实现此功能，但是我找不到在word2Vec中实现此目的的方法

解决方法

来自gensim documentation： 初始化模型，例如：

from gensim.test.utils import common_texts,get_tmpfile
from gensim.models import Word2Vec
    
path = get_tmpfile("word2vec.model")
    
model = Word2Vec(common_texts,size=100,window=5,min_count=1,workers=4)
model.save("word2vec.model")

现在，您可以通过以下方式获取例如word的单词向量：

model.wv['word']  # numpy vector of a word (OR: model.word_vec("word"))