为 CNN 文本分类保存 Word2Vec答案

【问题标题】：Saving Word2Vec for CNN Text Classification为 CNN 文本分类保存 Word2Vec
【发布时间】：2016-11-28 01:14:23
【问题描述】：

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

【问题讨论】：

【解决方案1】：

在tensorflow的情况下，不酸洗不是保存模型的方式。

Tensorflow 提供了 tensorflow 服务，用于将模型保存为 proto bufs（用于导出模型）。保存模型的方法是将 tensorflow 会话保存为： saver.save(sess, 'my_test_model',global_step=1000)

【讨论】：

【解决方案2】：

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

【讨论】：