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

Question

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

现在我们正在尝试使用 Saver 保存/恢复模型，但我们不知道在哪里使用 Saver API，因为 text_classification.py 中的代码根本不使用 Session，并且 Saver API 需要 session 来保存/恢复。

Answer 1

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

此外，您还可以在 constructor 中指定 model_dir :

Directory to save model parameters, graph and etc. This can also be used to load checkpoints from the directory into a estimator to continue training a previously saved model. If None, the model_dir in config will be used if set. If both are set, they must be same. If both are None, a temporary directory will be used.