python - 创建 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')

只有 1 个模型文件

test_model

我错了哪一部分？

Answer 1

Gensim 的本机仅在模型的某些部分大于某个阈值时.save()将其保存到此类单独的文件（如等）中。test_model.syn1neg.npy当它们很小时，它们会被“腌制”到单个模型保存文件中。

所以这里没有问题/错误。如果您开始使用更多单词训练更大的模型，您可能会看到那些其他文件重新出现。（当你这样做时，一定要把它们放在主test_model文件旁边，如果将它们复制/移动到其他地方——所有文件都需要一起重新load()模型。）