如何在 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"))
我正在尝试检索我训练的 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"))