Represents the output of a regression head.
Inherits From: ExportOutput
View aliases
Compat aliases for migration
See Migration guide for more details.
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
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.
|