如何导出estimator

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.

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部