From 587ec10386c3a939b4dbb7ead2d05516180d95c5 Mon Sep 17 00:00:00 2001 From: Yong Tang Date: Wed, 19 Sep 2018 03:33:51 +0000 Subject: Fix warning in text_classification_character_cnn.py Signed-off-by: Yong Tang --- tensorflow/examples/learn/text_classification_character_cnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tensorflow/examples') diff --git a/tensorflow/examples/learn/text_classification_character_cnn.py b/tensorflow/examples/learn/text_classification_character_cnn.py index afda170e2a..b8506fa8a4 100644 --- a/tensorflow/examples/learn/text_classification_character_cnn.py +++ b/tensorflow/examples/learn/text_classification_character_cnn.py @@ -74,7 +74,7 @@ def char_cnn_model(features, labels, mode): kernel_size=FILTER_SHAPE2, padding='VALID') # Max across each filter to get useful features for classification. - pool2 = tf.squeeze(tf.reduce_max(conv2, 1), squeeze_dims=[1]) + pool2 = tf.squeeze(tf.reduce_max(conv2, 1), axis=[1]) # Apply regular WX + B and classification. logits = tf.layers.dense(pool2, MAX_LABEL, activation=None) -- cgit v1.2.3