我想为我的文本语料库训练自己的Word2Vec模型。我可以从TensorFlow的教程中获取代码。我不知道的是如何保存此模型以便以后用于CNN文本分类?我应该使用pickle来保存它,然后再读它吗?
答案 0 :(得分:0)
您可以使用pickle将其保存到磁盘。然后,在创建CNN模型时,加载已保存的单词嵌入表,并使用它来初始化保存CNN分类器的单词嵌入的TensorFlow变量。
答案 1 :(得分:0)
在张量流的情况下,没有酸洗不是保存模型的方法。
Tensorflow提供tensorflow服务,用于将模型保存为proto bufs(用于导出模型)。保存模型的方法是将tensorflow会话保存为: saver.save(sess,'my_test_model',global_step = 1000)
以下是完整答案的链接: 的 Tensorflow: how to save/restore a model? 强>