diff options
Diffstat (limited to 'tensorflow/examples/learn/text_classification_character_rnn.py')
-rw-r--r-- | tensorflow/examples/learn/text_classification_character_rnn.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/examples/learn/text_classification_character_rnn.py b/tensorflow/examples/learn/text_classification_character_rnn.py index 86adc056ad..15733821fb 100644 --- a/tensorflow/examples/learn/text_classification_character_rnn.py +++ b/tensorflow/examples/learn/text_classification_character_rnn.py @@ -59,9 +59,7 @@ def char_rnn_model(features, labels, mode): 'prob': tf.nn.softmax(logits) }) - onehot_labels = tf.one_hot(labels, MAX_LABEL, 1, 0) - loss = tf.losses.softmax_cross_entropy( - onehot_labels=onehot_labels, logits=logits) + loss = tf.losses.sparse_softmax_cross_entropy(labels=labels, logits=logits) if mode == tf.estimator.ModeKeys.TRAIN: optimizer = tf.train.AdamOptimizer(learning_rate=0.01) train_op = optimizer.minimize(loss, global_step=tf.train.get_global_step()) |