如何停止Keras显示“正在使用XXX后端”?

蔡易翔*_*蔡易翔 5 python keras

导入keras时,将显示“使用XXX后端”。

import keras
Run Code Online (Sandbox Code Playgroud)

输出:

Using TensorFlow backend.
Run Code Online (Sandbox Code Playgroud)

但是我清楚知道我在用什么。

如何禁用它?

蔡易翔*_*蔡易翔 8

只需使用下面的代码。

import os
import sys
stderr = sys.stderr
sys.stderr = open(os.devnull, 'w')
import keras
sys.stderr = stderr
Run Code Online (Sandbox Code Playgroud)


mau*_*omi 0

您有两个选择:首先在此处注释keras/backend/__init__.pystring print('Using TensorFlow backend.'),但当然不建议使用此选项,因为您将在 keras 模块中编辑代码。第二个选项是:

import sys
stdout = sys.stdout
sys.stdout = open('/dev/null', 'w')
import keras
sys.stdout = stdout
Run Code Online (Sandbox Code Playgroud)

它并不优雅,但很有效。