View source on GitHub |
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. |