在 Ipython Notebook 中将 keras 后端更改为 Theano

Question

我确实keras.json按照Keras 文档页面上的说明更改了文件。但是在我的 Ipython 笔记本中，它仍然说我使用 Tensorflow 作为后端。

也许它以某种方式与 Jupyter 设置有关？请帮助。我什至不知道如何找出问题出在哪里。谢谢！

Answer 1

您可以在笔记本的开头尝试以下操作：

import os
os.environ["KERAS_BACKEND"] = "theano"
import keras; import keras.backend
if keras.backend.backend() != 'theano':
    raise BaseException("This script uses other backend")
else:
    keras.backend.set_image_dim_ordering('th')
    print("Backend ok")

基本上环境 KERAS_BACKEND 可以在某个时候被 Jupyter 覆盖，所以这是在导入 keras.backend 之前强制它成为某些东西的一种方法。