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