阅读(11k) 书签 (0)

TensorFlow函数:tf.estimator.Exporter

2018-05-02 09:58 更新

tf.estimator.Exporter函数

Exporter类

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

一个代表模型导出类型的类.

属性

  • name

    目录名称.

    该类型导出的导出基目录下的目录名称,不应该是None,也不是空的.

方法

export

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

将给定Estimator导出为特定的格式.

函数参数:

  • estimator:要导出的Estimator.
  • export_path:一个字符串,其中包含要编写导出的目录.
  • checkpoint_path:要导出的检查点路径.
  • eval_result:这个检查点的Estimator.evaluate输出.
  • is_the_final_export:当这是训练结束时的导出时,此布尔值为True;训练期间的中间出口为False.如果TrainSpec.max_steps是None,则当向tf.estimator.train_and_evaluate is_the_final_export传递Exporter时,总是为False.

返回值:

tf.estimator.Exporter函数返回导出目录的字符串路径,如果跳过导出,则为None.