-
前提條件
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)