保存Word2Vec用于CNN文本分类

时间:2016-07-24 18:06:16

标签: tensorflow deep-learning text-classification word2vec

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

2 个答案:

答案 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?