W3cubDocs

/TensorFlow 2.4

Module: tf.estimator.export

All public utility methods for exporting Estimator to SavedModel.

This file includes functions and constants from core (model_utils) and export.py

Classes

class ClassificationOutput: Represents the output of a classification head.

class ExportOutput: Represents an output of a model that can be served.

class PredictOutput: Represents the output of a generic prediction head.

class RegressionOutput: Represents the output of a regression head.

class ServingInputReceiver: A return type for a serving_input_receiver_fn.

class TensorServingInputReceiver: A return type for a serving_input_receiver_fn.

Functions

build_parsing_serving_input_receiver_fn(...): Build a serving_input_receiver_fn expecting fed tf.Examples.

build_raw_serving_input_receiver_fn(...): Build a serving_input_receiver_fn expecting feature Tensors.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/estimator/export