gensim - 如何从 gensim Doc2Vec 获取词向量?

标签 gensim word2vec doc2vec

我训练了一个 gensim.models.doc2vec.Doc2Vec 模型
d2v_model = Doc2Vec(sentences, size=100, window=8, min_count=5, workers=4)
我可以通过以下方式获取文档向量
docvec = d2v_model.docvecs[0]

如何从训练有素的模型中获取词向量?

最佳答案

Doc2Vec 继承自 Word2Vec,因此您可以直接通过索引模型来访问与 Word2Vec 中相同的词向量:

wv = d2v_model['apple']

但是请注意,像纯 DBOW ( dm=0 ) 这样的 Doc2Vec 训练模式不需要或创建词向量。 (对于许多目的,纯 DBOW 仍然可以很好和快速地工作!)如果您确实从这样的模型访问词向量,它们将只是自动随机初始化的向量,没有任何意义。

仅当 Doc2Vec 模式本身共同训练词向量时,如在 DM 模式(默认 dm=1 )或将可选词训练添加到 DBOW( dm=0, dbow_words=1 )时,词向量和文档向量同时学习.

关于gensim - 如何从 gensim Doc2Vec 获取词向量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37335842/

相关文章:

python - 使用 gensim 将 LDA 应用于语料库进行训练

neural-network - 如何从 gensim word2vec 模型推断新词向量?

python - 是否可以从 python 中的句子语料库重新训练 word2vec 模型(例如 GoogleNews-vectors-negative300.bin)?

python - Gensim word2vec 增强或合并预训练向量

python - 使用语义词表示(例如 word2vec)来构建分类器

python - 如何加载预训练的 doc2vec 模型并使用它的向量

python - Word2Vec 和 Gensim 参数等价

machine-learning - 如何使用 Word2Vec 获取单词列表的向量?

python - Doc2Vec找到相似的句子

python - doc2vec - 如何更快地推断文档向量?