从 gensim word2Vec 获取权重矩阵答案

【问题标题】：Get weight matrices from gensim word2Vec从 gensim word2Vec 获取权重矩阵
【发布时间】：2017-04-30 23:31:15
【问题描述】：

我在 python 中使用 gensim word2vec 包。我想检索在skip-gram学习过程中学习到的W和W'权重矩阵。

在我看来，model.syn0 给了我第一个，但我不确定如何获得另一个。有什么想法吗？

我真的很想找到任何关于模型可访问属性的详尽文档，因为官方文档似乎并不准确（例如，syn0 没有被描述为属性）

【问题讨论】：

我刚刚尝试了一个小数据集，它给了我整个嵌入矩阵np.shape(model.syn0)

标签： python machine-learning nlp word2vec gensim

【解决方案1】：

model.wv.syn0 包含输入嵌入矩阵。输出嵌入在使用 hierarchical softmax (hs=1) 训练时存储在 model.syn1 中，或者在使用负采样 (negative>0) 时存储在 model.syn1neg 中。而已！在不启用分层 softmax 和负采样时，Word2Vec 使用单个权重矩阵 model.wv.syn0 进行训练。

另见相关讨论here。

【讨论】：