0

我已经成功地（我希望）使用 tf.Estimator 训练和评估了一个模型，我达到了大约 83-85% 的训练/评估准确度。所以现在，我想使用 Estimator 类中的 predict() 函数调用在单独的数据集上测试我的模型。最好我想在单独的脚本中执行此操作。

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

4

1 回答 1

1

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

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

于 2017-12-13T15:20:21.320 回答