如何在 Word2Vec 中检索数组

How to retrieve array in Word2Vec

我正在尝试检索我训练的 word2vec 模型中某个单词的 array/vector。在 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"))