创建word2vec模型syn1neg.npy扩展名

Question

在创建模型时,没有更多具有扩展完成的模型

.syn1neg.npy

syn0.npy

我的代码如下:

corpus= x+y
tok_corp= [nltk.word_tokenize(sent.decode('utf-8')) for sent in corpus]
model = gensim.models.Word2Vec(tok_corp, min_count=1, size = 32)
model.save('/home/Desktop/test_model')

model = gensim.models.Word2Vec.load('/home/kafein/Desktop/chatbot/test_model')

只有一个模型文件

test_model

哪一部分我错了？

Answer 1

如果Gensim的原生大于特定阈值,.save()它只会将模型的某些部分保存到这些单独的文件(如test_model.syn1neg.npy等)中.当它们很小时,它们会被"腌制"到单个模型保存文件中.

所以这里没有问题/错误.如果您开始使用更多单词训练更大的模型,您可能会看到其他文件重新出现.(当你这样做时,一定要将它们与主test_model文件放在一起,如果将它们复制/移动到其他地方 - 需要将所有文件放在一起才能重新load()建立模型.)