tf.contrib.layers.one_hot_encoding
Transform numeric labels into onehot_labels using tf.one_hot
.
tf.contrib.layers.one_hot_encoding( labels, num_classes, on_value=1.0, off_value=0.0, outputs_collections=None, scope=None )
Args | |
---|---|
labels | [batch_size] target labels. |
num_classes | Total number of classes. |
on_value | A scalar defining the on-value. |
off_value | A scalar defining the off-value. |
outputs_collections | Collection to add the outputs. |
scope | Optional scope for name_scope. |
Returns | |
---|---|
One-hot encoding of the labels. |
© 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/one_hot_encoding