2016-07-24 35 views
0

我想为我的文本语料库训练自己的Word2Vec模型。我可以从TensorFlow的教程中获取代码。我不知道如何保存这个模型用于以后的CNN文本分类?我应该用腌菜来保存它,然后再读它吗?将Word2Vec保存为CNN文本分类

回答

0

您可以使用pickle将其保存到磁盘。然后,在创建CNN模型时,加载已保存的单词嵌入表并使用它来初始化包含CNN分类器的单词嵌入的TensorFlow变量。

0

没有酸洗不是保存张量流动模型的方式。

Tensorflow提供张量流服务,用于将模型保存为原始bufs(用于导出模型)。保存模型的方法是保存tensorflow会话: saver.save(SESS, 'my_test_model',global_step = 1000)

继承人的链接,完整的答案: Tensorflow: how to save/restore a model?