2016-12-15 98 views
7

我在python中使用gensim word2vec包。 我想檢索在跳躍學習期間學過的WW'權重矩陣。從gensim獲取重量矩陣word2Vec

在我看來,model.syn0給了我第一個,但我不知道我怎麼能得到另一個。任何想法?

我真的愛找對模型訪問屬性的任何詳盡的文件,因爲官方的一個似乎不準確(例如syn0沒有被描述爲一個屬性)

+1

我剛試過一個小數據集,它給了我整個嵌入矩陣'np.shape(model.syn0)' –

回答

2

model.wv.syn0包含輸入嵌入矩陣。 輸出使用hierarchical softmaxhs=1)或model.syn1neg(使用負樣本(negative>0))時,嵌入存儲在model.syn1中。而已!當分層softmax和負採樣都未啓用時,Word2Vec使用單一權重矩陣model.wv.syn0進行培訓。

另請參閱相關討論here