我正在用不同的数据训练我自己的word2vec模型。为了将得到的模型实现到我的分类器中,并将结果与原始的预先训练的Word2vec模型进行比较,我需要将该模型保存在binary extension.bin中。这是我的代码,句子是一个短消息列表。import gensim, logging
logging.basicConfig(format='%(asctime)s : %(levelname)s : %(message)s', level=logging.INFO)
sentences = gensim.models.word2vec.LineSentence('dati.txt')
model = gensim.models.Word2Vec(
sentences, size=300, window=5, min_count=5, workers=5,
sg=1, hs=1, negative=0
)
model.save_word2vec_format('model.bin', binary=True)
最后一个方法save_word2vec_format给出了这个错误:
AttributeError: 'Word2Vec' object has no attribute 'save_word2vec_format'
我错过了什么?我看过gensim和其他论坛的文档。这个repo on github使用几乎相同的配置,所以我无法理解出了什么问题。我试图从skipgram切换到cbow,从分层softmax切换到负采样,但没有结果。
提前谢谢你!