阅读(3948) (1)

如何导出estimator

2017-09-29 17:16:24 更新

tf.estimator.export

定义在:tensorflow/python/estimator/export/export_lib.py.

tf.estimator.export 是用来导出 estimator 的,并且该方法很实用.

class ClassificationOutput:表示分类头的输出.

class ExportOutput:表示可以服务的模型的输出.

class PredictOutput:表示通用预测头的输出.

class RegressionOutput:表示回归头的输出.

class ServingInputReceiver:一个 serve_input_receiver_fn 的返回类型.

函数

build_parsing_serving_input_receiver_fn(...):构建一个 expect_input_receiver_fn 以提供 tf.Examples.

build_raw_serving_input_receiver_fn(...):构建一个 serve_input_receiver_fn 期望的 feature Tensors.