2016-07-24 63 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?