contrib.layers.one_hot_encoding
tf.contrib.layers.one_hot_encoding
tf.contrib.layers.one_hot_encoding
one_hot_encoding( labels, num_classes, on_value=1.0, off_value=0.0, outputs_collections=None, scope=None )
Defined in tensorflow/contrib/layers/python/layers/layers.py
.
See the guide: Layers (contrib) > Higher level ops for building neural network layers
Transform numeric labels into onehot_labels using tf.one_hot
.
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.
© 2017 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/api_docs/python/tf/contrib/layers/one_hot_encoding