阅读(10.1k) 书签 (0)

TensorFlow函数:tf.estimator.LatestExporter

2018-05-04 10:08 更新

tf.estimator.LatestExporter函数

LatestExporter类

继承自:Exporter

定义在:tensorflow/python/estimator/exporter.py.

LatestExporter类定期导出服务图和检查点.

除了导出之外,这个类还会收集过时的导出.

属性

name

方法

__init__

__init__(
    name,
    serving_input_receiver_fn,
    assets_extra=None,
    as_text=False,
    exports_to_keep=5
)

创建一个用于tf.estimator.EvalSpec的Exporter.

参数:

  • name:将在导出路径中使用的Exporter的唯一名称.
  • serving_input_receiver_fn:不带参数并返回一个ServingInputReceiver的函数.
  • assets_extra:一个可选的字典,指定如何填充导出的SavedModel中的assets.extra目录.每个密钥都应该提供相对于assets.extra目录的目标路径(包括文件名).相应的值给出了要复制的源文件的完整路径.例如,复制单个文件而不重命名它的简单情况被指定为{'my_asset_file.txt': '/path/to/my_asset_file.txt'}.
  • as_text:是否以文本格式编写SavedModel原型,默认为False.
  • exports_to_keep:要保留的导出数量.旧的导出将被收集;默认为5;设置None以禁用回收.

可能引发的异常:

  • ValueError:如果任何参数无效,则会引发此异常.

export

export(
    estimator,
    export_path,
    checkpoint_path,
    eval_result,
    is_the_final_export
)