diff options
author | Yong Tang <yong.tang.github@outlook.com> | 2018-09-19 03:33:51 +0000 |
---|---|---|
committer | Yong Tang <yong.tang.github@outlook.com> | 2018-09-19 03:34:42 +0000 |
commit | 587ec10386c3a939b4dbb7ead2d05516180d95c5 (patch) | |
tree | b6ea9b9d2d248a27a5de8c27194d4c4828332df0 /tensorflow/examples | |
parent | 9fe177881224571aff0c267593f747f5fd7a2967 (diff) |
Fix warning in text_classification_character_cnn.py
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/examples')
-rw-r--r-- | tensorflow/examples/learn/text_classification_character_cnn.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |