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.