我在尝试从我训练的word2vec模型中获取一个词的数组/向量。在SpaCy中,可以通过model.vocab.get_vector(“word”)来实现,但我在word2Vec中找不到相应的方法。
回答:
根据gensim文档的说明,可以通过以下方式初始化模型:
from gensim.test.utils import common_texts, get_tmpfilefrom 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向量(或者:model.word_vec("word"))