当前位置:首页 >> 脚本专栏

Keras 切换后端方式(Theano和TensorFlow)

实验室新装了keras,发现keras默认后端是tensorflow,想换回theano,看了官方文档也没搞懂,最终搞定,很简单。

中文文档的描述:keras中文文档,切换后端

其实就是在C:\Users\75538(75538是我的windos用户名字,找你对应的用户名就行)下有个文件夹.keras,里面有keras.json文件,改一下里面的内容就好了,如果没有文件夹和文件,手动创建就行。

用theano的话,keras.json写入

{
 "image_dim_ordering": "th", 
 "epsilon": 1e-07, 
 "floatx": "float32", 
 "backend": "theano"
}

用thesorflow的话,keras.json写入

{
 "image_dim_ordering": "tf", 
 "epsilon": 1e-07, 
 "floatx": "float32", 
 "backend": "tensorflow"
}

最后保存就可以了

补充知识:anaconda2-keras安装;keras后端修改

一、anaconda2-keras安装

在anaconda2平台下,安装keras库。记录一下,主要是我一开始觉得很简单,pip一下就行了,结果我用pip install keras安装时,就遇到了这样一个问题:ImportError: cannot import name np_utils。

解决方案:

That tutorial was written on June 9th, 2016.Keras 2 was released in March 2017. Try installing the old version, using pip installkeras==1.2.2.

总结一下就是:

1.先装上anaconda2_python2.7

2.再在anancona prompt中用 pip install keras==1.2.2 装Keras

3.spyder中输入 import Keras,检查是否安装成功

二、keras后端修改

代码用的后端是'theano',故要修改后端

将C:\Anaconda2\Lib\site-packages\keras\backend\__init__.py的line 27修改

# Default backend: TensorFlow.
#_BACKEND = 'tensorflow'
_BACKEND = 'theano'

然后,python-> import keras,检查是否修改成功。

以上这篇Keras 切换后端方式(Theano和TensorFlow)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。