首页 > 技术文章 > Keras简介

xmeo 2017-03-16 19:08 原文

Keras 是一个兼容 Theano 和 Tensorflow 的神经网络高级包, 可以用他来更加快速组件一个神经网络

Backend

Keras 可以基于两个Backend,一个是 Theano,一个是 Tensorflow。

如何查看用的是那个backend

当我们import keras后,运行时,终端会打印

Using TensorFlow backend.

 

这说明我们目前用的是tensorflow的backend

如何修改backend

import keras
import os
os.environ['KERAS_BACKEND']='theano'

import os
os.environ['KERAS_BACKEND']='tensorflow'

 

推荐阅读