通过SessionRunHooks访问估计器评估结果
通过SessionRunHooks访问估计器评估结果
提问于 2018-10-04 12:57:12
我的一个想法是创建一个继承自SessionRunHook的类，在after_run方法中执行我想要的工作。根据documentation，我可以使用before_run指定传递给after_run的内容。但是，我找不到从传递给before_run的信息中访问我想要的评估指标结果的方法。
我查看了Estimator代码，它似乎是writing the results to a summary file，所以我的另一个想法是在after_run方法中读回这段代码，但summary api似乎没有提供任何读取操作。
有没有其他方法可以实现我想要做的事情？不使用Estimator类是不可取的，因为这将涉及到对我正在使用的代码进行重大更改。
回答 1
Stack Overflow用户
回答已采纳
发布于 2018-10-15 06:21:38
检查点是关于故障恢复的，涉及保存完整的训练状态(权重、全局步数等)。
导出的模型将写入名为“exporter”的目录，服务输入功能指定最终用户将向预测服务提供什么。
您可以使用"Best Exporter“类仅导出性能最佳的模型：
https://www.tensorflow.org/api_docs/python/tf/estimator/BestExporter
这个类导出了最佳模型的服务图和检查点。
此外，每当新模型优于任何现有模型时，它都会执行模型导出。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/52639289
复制相关文章
相似问题