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)

你可能感兴趣的:(机器学习,NLP)