Word2Vec保存词向量的方法

1.以model.save()方法保存词向量
保存词向量
‘’’
import gensim
model = gensim.models.Word2Vec(documents, size=300)
model.train(documents, total_examples=len(documents), epochs=10)
model.save("…/input/Word2vec.w2v")
‘’’
加载词向量
‘’’
import gensim
word2vec = gensim.models.word2vec.Word2Vec.load("./input/Quora.w2v").wv
‘’’

2.保存为二进制的词向量
保存词向量
‘’’
model.wv.save_Word2Vec_format(embedding_path,binary=True)
model.wv.save_Word2Vec_format(embedding_path,binary=False)非二进制
‘’’
加载词向量
‘’’
import gensim
word2vec = gensim.models.KeyedVectors.load_word2vec_format(embedding_path,binary=True)
‘’’

3.使用numpy进行保存和加载
保存数组数据的文件可以是二进制格式或者文本格式，二进制格式的文件可以是Numpy专用的二进制类型和无格式类型。

使用np.save()保存npy文件，np.load()加载npy文件。

作者：top小酱油
链接：https://www.jianshu.com/p/fbce9c8f5536
来源：简书
简书著作权归作者所有，任何形式的转载都请联系作者获得授权并注明出处。