tf.contrib.layers.multi_class_target

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

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/layers/multi_class_target