from gensim.models import word2vec
sentences = word2vec.Text8Corpus('TextFile')
model = word2vec.Word2Vec(sentences, size=200, min_count = 2, workers = 4)
print model['king']
输出向量是“king”的上下文向量还是词嵌入向量?如何同时获取“king”的上下文向量和词嵌入向量?谢谢!
from gensim.models import word2vec
sentences = word2vec.Text8Corpus('TextFile')
model = word2vec.Word2Vec(sentences, size=200, min_count = 2, workers = 4)
print model['king']
输出向量是“king”的上下文向量还是词嵌入向量?如何同时获取“king”的上下文向量和词嵌入向量?谢谢!
这是关于“king”的嵌入向量。
如果使用分层softmax,上下文向量如下:
model.syn1
如果您使用负采样,它们就是:
model.syn1neg
向量可以通过以下方式访问:
model.syn1[model.vocab[word].index]