Tensorflow Estimator：在单独的脚本中使用predict（）函数

Question

我已成功（我希望）使用tf.Estimator训练和评估模型，我的火车/评估精度达到83-85％左右。所以现在，我想使用Estimator类中的predict（）函数调用在单独的数据集上测试我的模型。我希望在一个单独的脚本中这样做。

我在this说我需要导出为SavedModel，但这真的有必要吗？查看Estimator类的documentation，似乎我可以通过model_dir参数将路径传递给我的检查点和图形文件。有没有人有这方面的经验？当我在用于验证的同一数据集上运行我的模型时，我没有获得与验证阶段相同的性能...： - （

Answer 1

我认为您只需要一个包含model_fn定义的单独文件。比在另一个脚本中使用相同的model_fn定义和相同的model_dir实例化相同的估算器类。

这是有效的，因为Estimator API会自行恢复tf.Graph个定义和最新的model.ckpt文件，因此您可以继续进行培训，评估和预测。