Missed TensorFlow World? Check out the recap. Learn more

tf.estimator.export.RegressionOutput

TensorFlow 2.0 version View source on GitHub

Class RegressionOutput

Represents the output of a regression head.

Inherits From: ExportOutput

Aliases:

  • Class tf.compat.v1.estimator.export.RegressionOutput
  • Class tf.compat.v2.estimator.export.RegressionOutput

__init__

View source

__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.

Properties

value

Methods

as_signature_def

View source

as_signature_def(receiver_tensors)