tf.contrib.layers.multi_class_target
Creates a _TargetColumn for multi class single label classification. (deprecated)
tf.contrib.layers.multi_class_target( n_classes, label_name=None, weight_column_name=None )
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