我们是否可以使用 tf 2.0 将训练好的 Gensim Word2Vec 模型保存为保存的模型?tf.saved_model.save
? 换句话说,我如何将训练好的嵌入向量保存为保存的模型签名,以便与 tensorflow 2.0 一起使用。下面的步骤正常不正确。
model = gensim.models.Word2Vec(...)
model.init_sims(..)
model.train(..)
model.save(..)
module = gensim.models.KeyedVectors.load_word2vec(...)
tf.saved_model.save(
module,
export_dir
)
EDIT:
这个例子帮助我了解了如何做到这一点。https:/keras.ioexamplesnlppretrained_word_embeddings