如何屏蔽TensorFlow运行结果中显示的警告信息

这个警告本质是TFlearn库的内部代码还在使用TensorFlow已经标记为废弃的keep_dims参数，而TensorFlow现在推荐使用小写的keepdims（注意最后没有下划线）。你的代码功能本身不受影响，但如果看着警告不舒服，可以用下面几种方式解决：

方法1：升级TFlearn到最新版本 #

最省心的办法是直接更新TFlearn，新版本的库应该已经修复了这个参数兼容问题。打开终端执行：

pip install --upgrade tflearn

更新后重新运行代码，警告应该就消失了。

方法2：手动修改TFlearn源码（适合暂时无法升级的情况） #

如果因为环境限制没法升级，你可以直接修改报错文件里的代码：

• 找到路径C:\Users\vsureshx079451\AppData\Local\Programs\Python\Python36\lib\site-packages\tflearn\objectives.py
• 打开这个文件，定位到第66行
• 把代码里的keep_dims替换成keepdims（比如原来的reduce_sum(..., keep_dims=True)改成reduce_sum(..., keepdims=True)）
• 保存文件后重新运行代码，警告就会消失

方法3：临时抑制TensorFlow警告（不推荐，除非紧急情况） #

如果只是不想看到警告，但不想修改任何东西，可以在代码开头添加以下代码来屏蔽TensorFlow的警告输出：

import tensorflow as tf
tf.get_logger().setLevel('ERROR')

不过这种方式会把所有TensorFlow的警告都屏蔽，可能会错过其他重要的提示信息，所以优先推荐前两种方法。

内容的提问来源于stack exchange，提问作者vinothvs