tf.estimator.export.RegressionOutput

Class RegressionOutput

Inherits From: ExportOutput

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

Represents the output of a regression head.

Properties

value

Methods

__init__

__init__(value)

Constructor for RegressionOutput.

Args:

  • value: a float Tensor giving the predicted values. Required.

Raises:

  • ValueError: if the value is not a Tensor with dtype tf.float32.

as_signature_def

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.