首页 文章

保存Word2Vec用于CNN文本分类

提问于
浏览
0

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

2 回答

  • 0

    在张量流量的情况下,没有酸洗不是保存模型的方法 .

    Tensorflow提供tensorflow用于将模型保存为proto bufs(用于导出模型) . 保存模型的方法是将tensorflow会话保存为: saver.save(sess, 'my_test_model',global_step=1000)

    下面是完整答案的链接: Tensorflow: how to save/restore a model?

  • 0

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

相关问题