从tensorflow tensorflow / examples / learn / text_classification.py保存示例文本分类模型

时间:2017-10-12 12:50:17

标签: machine-learning tensorflow text-classification

我们正在尝试使用TensorFlow示例中的文本分类示例(tensorflow / examples / learn / text_classification.py)。它适用于db_pedia数据。

现在我们正在尝试使用Saver保存/恢复模型,但是我们没有在哪里使用Saver API,因为text_classification.py中的代码根本不使用Session,Saver API需要会话来保存/恢复。

1 个答案:

答案 0 :(得分:0)

此示例使用tf.estimator.Estimator,它有一个特殊方法 export_savedmodel用于保存。

此外,您可以在constructor中指定model_dir

  

保存模型参数,图表等的目录。这也可以   用于将目录中的检查点加载到估算器中   继续训练以前保存的模型。如果为None,则为model_dir   如果设置,将使用config。如果两者都设置,它们必须相同。如果   两者都是None,将使用临时目录。