如何导出estimator

由 Carrie 创建, 最后一次修改 2017-09-29

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.

以上内容是否对您有帮助:
二维码
建议反馈
二维码