我们正在尝试使用TensorFlow示例中的文本分类示例(tensorflow / examples / learn / text_classification.py)。它与db_pedia数据一起很好地工作。
现在,我们尝试使用Saver保存/恢复模型,但是我们没有在哪里使用Saver API作为text_classification中的代码。py根本不使用Session,并且Saver API需要使用session来保存/恢复。
最佳答案
本示例使用tf.estimator.Estimator
,它具有特殊的方法export_savedmodel
保存。
另外,您可以在constructor中指定model_dir
:
保存模型参数,图形等的目录。也可以是
用于将目录中的检查点加载到估算器中
继续训练以前保存的模型。如果为None,则model_dir在
如果设置,将使用config。如果两者都设置,则必须相同。如果
两者均为“无”,将使用一个临时目录。