使用函数:model.wv.index2word()
废话不多说,我们来看代码,假设我们已经训练好了模型。
from gensim.models.word2vec import Word2Vec
import pickle
model = Word2Vec.load('./all_data/WORD_MODEL') #WORD_MODEL是我已经生成的模型
print(model.wv.index2word()) #获得所有的词汇
for word in model.wv.index2word():
print(word,model[word]) #获得词汇及其对应的向量