def _maybe_export(self, eval_result): # pylint: disable=unused-argument
"""Export the Estimator using export_fn, if defined."""
export_dir_base = os.path.join(
compat.as_bytes(self._estimator.model_dir),
compat.as_bytes("export"))
export_results = []
for strategy in self._export_strategies:
# TODO(soergel): possibly, allow users to decide whether to export here
# based on the eval_result (e.g., to keep the best export).
export_results.append(
strategy.export(
self._estimator,
os.path.join(
compat.as_bytes(export_dir_base),
compat.as_bytes(strategy.name))))
return export_results
experiment.py 文件源码
python
阅读 34
收藏 0
点赞 0
评论 0
评论列表
文章目录