如何保存和载入手套模型?

问题描述 投票:4回答:2

我使用Python 3.5做我的研究。我想利用手套字的嵌入的。我如何保存和glove.fit后装入我的手套模型?我已经编写它像这样

glove.fit(corpus.matrix,epochs=1,no_threads=4,verbose=True)
glove.save('glove.model')
python-3.x word-embedding
2个回答
0
投票

检查此here

现在,让你的数据训练后,使用此:

from gensim.scripts.glove2word2vec import glove2word2vec
glove2word2vec(glove_input_file=file, word2vec_output_file="gensim_glove_vectors.txt")    
from gensim.models.keyedvectors import KeyedVectors
model = KeyedVectors.load_word2vec_format("gensim_glove_vectors.txt", binary=False)

之后,您可以为您使用gensim模型只是用这个。例如,

print("Similarity between {} and {} is {}".format(word1,word2,model.wv.similarity(word1, word2)))
print("Most similar words to {} are :{}\n".format(word1,model.most_similar(positive=[word1],topn=10)))

0
投票
from gensim.models import KeyedVectors
# load the Stanford GloVe model
model = KeyedVectors.load_word2vec_format(filename, binary=False)

如果你的模型包含在变量“模式”

您可以保存这样的模式:

model.save('model.bin')

您可以加载像这样保存的模型:

new_model = KeyedVectors.load('model.bin')

您现在可以使用所加载的模型:

result = new_model.most_similar(positive=['woman', 'king'], negative=['man'], topn=1)
© www.soinside.com 2019 - 2024. All rights reserved.