我在python中使用gensim word2vec包。
我想检索在跳过学习过程中学习过的W
和W'
权重矩阵。
在我看来,model.syn0
给了我第一个,但我不确定如何能得到另一个。有什么想法吗?
我真的很想找到关于模型可访问属性的任何详尽文档,因为官方文档看起来并不精确(例如syn0没有被描述为属性)
答案 0 :(得分:5)
model.wv.syn0
包含输入嵌入矩阵。 输出嵌入在使用hierarchical softmax(model.syn1
)进行培训时存储在hs=1
中,或者在使用负抽样时存储在model.syn1neg
中negative>0
})。而已!如果未启用分层softmax和负抽样,Word2Vec
会使用单个权重矩阵model.wv.syn0
进行培训。
另见相关讨论here。