从gensim word2Vec获取权重矩阵

时间:2016-12-15 11:19:12

标签: python machine-learning nlp word2vec gensim

我在python中使用gensim word2vec包。 我想检索在跳过学习过程中学习过的WW'权重矩阵。

在我看来,model.syn0给了我第一个,但我不确定如何能得到另一个。有什么想法吗?

我真的很想找到关于模型可访问属性的任何详尽文档,因为官方文档看起来并不精确(例如syn0没有被描述为属性)

1 个答案:

答案 0 :(得分:5)

model.wv.syn0包含输入嵌入矩阵。 输出嵌入在使用hierarchical softmaxmodel.syn1)进行培训时存储在hs=1中,或者在使用负抽样时存储在model.syn1negnegative>0 })。而已!如果未启用分层softmax和负抽样,Word2Vec会使用单个权重矩阵model.wv.syn0进行培训。

另见相关讨论here