Module: tf.estimator.export

Defined in tensorflow/python/estimator/export/export_lib.py.

Utility methods for exporting Estimator.

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.

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.