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