parameters: (self, label_dimension, weight_column=None)
docstring: `Head` for regression.
docstring_summary: `Head` for regression.
docstring_tokens: Head for regression .
function: def __init__(self, label_dimension, weight_column=None):
      """`Head` for regression."""
      if label_dimension < 1:
        raise ValueError('Invalid label_dimension %s.' % label_dimension)
      self._logits_dimension = label_dimension
      self._weight_column = weight_column
function_tokens: def __init__ ( self , label_dimension , weight_column = None ) : if label_dimension < 1 : raise ValueError ( 'Invalid label_dimension %s.' % label_dimension ) self . _logits_dimension = label_dimension self . _weight_column = weight_column