Apply to speak at TensorFlow World. Deadline April 23rd. Propose talk

tf.contrib.layers.multi_class_target

tf.contrib.layers.multi_class_target(
    n_classes,
    label_name=None,
    weight_column_name=None
)

Defined in tensorflow/contrib/layers/python/layers/target_column.py.

Creates a _TargetColumn for multi class single label classification. (deprecated)

The target column uses softmax cross entropy loss.

Args:

  • n_classes: Integer, number of classes, must be >= 2
  • 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.

Returns:

An instance of _MultiClassTargetColumn.

Raises:

  • ValueError: if n_classes is < 2