如何导出estimator
2018-09-29 17:30 更新
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.