tf.estimator.export.RegressionOutput

View source on GitHub

Represents the output of a regression head.

Inherits From: ExportOutput

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.estimator.export.RegressionOutput, `tf.compat.v2.estimator.export.RegressionOutput`

tf.estimator.export.RegressionOutput(
    value
)

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

as_signature_def(
    receiver_tensors
)

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.