tf.estimator.export.RegressionOutput

View source on GitHub

Represents the output of a regression head.

Inherits From: ExportOutput

Args
value a float Tensor giving the predicted values. Required.
Raises
ValueError if the value is not a Tensor with dtype tf.float32.
Attributes
value

Methods

as_signature_def

View source

Generate a SignatureDef proto for inclusion in a MetaGraphDef.

The SignatureDef will specify outputs as described in this ExportOutput, and will use the provided receiver_tensors as inputs.

Args
receiver_tensors a Tensor, or a dict of string to Tensor, specifying input nodes that will be fed.

© 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/r1.15/api_docs/python/tf/estimator/export/RegressionOutput