从 gensim word2Vec 获取权重矩阵

Question

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

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

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

Answer 1

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

另请参阅相关讨论 。