word2vec 獲得當前的所有詞向量表的幾種方式

  • 前提條件

       1. 安裝gensim包
       2. 已經生成詞向量模型
    

在實際項目中,需要構建詞和詞向量字典表用於訓練,以下是我所用到的方式

方式1:

from gensim.models.word2vec import Word2Vec
model = Word2Vec.load(‘…/model/w2v_model’) #w2v_model是已經生成的模型

print(model.wv.index2word()) #獲得所有的詞彙
for word in model.wv.index2word():
print(word,model.wv[word]) #獲得詞彙及其對應的向量

方式2:

import numpy as np
from gensim.models.word2vec import Word2Vec
model = Word2Vec.load(’…/model/w2v_model’)
for word in model.wv.vocab.keys():
vec_string = np.array2string(model.wv[word]).replace(’[ ‘,’’).replace(’]’,’’).replace(’[’,’’).replace(’\n’,’’)
line = “{0} {1}\n”.format(word, vec_string)
print(line)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章