Gensim加载word2vec模型与简易使用

二进制文件

from gensim.models import KeyedVectors


word_model = KeyedVectors.load_word2vec_format('GoogleNews-vectors-negative300.bin.gz', binary=True)
word_model.has_index_for('123') # 判断词向量中是否包含 '123'
word_model.get_vector('word') # 打印word的向量

非二进制文件

from gensim.models import Word2Vec


word_model = Word2Vec.load('wiki_word2vec_model')
word_model.wv.has_index_for('123') # 判断词向量中是否包含 '123'
word_model.get_vector('word') # 打印word的向量

参考资料

Gensim训练模型 词向量的保存与调用：https://juejin.cn/post/6844903912256831502