TensorFlow 2.0 Beta is available Learn more

tf.contrib.layers.regression_target

View source on GitHub

Creates a _TargetColumn for linear regression. (deprecated)

tf.contrib.layers.regression_target(
    label_name=None,
    weight_column_name=None,
    label_dimension=1
)

Args:

  • label_name: String, name of the key in label dict. Can be null if label is a tensor (single headed models).
  • weight_column_name: A string defining feature column name representing weights. It is used to down weight or boost examples during training. It will be multiplied by the loss of the example.
  • label_dimension: dimension of the target for multilabels.

Returns:

An instance of _TargetColumn