用Keras将theano后端转换为tensorflow后端需要什么?

时间:2019-03-26 15:26:14

标签: tensorflow keras

我正在尝试转换https://github.com/hexiangnan/neural_collaborative_filtering

该库使用theano作为后端,将其更改为tensorflow后端有多大作用?
它也使用keras

搜索import theano.tensor as Ttheano时,我只会看到import

像将import语句更改为tensorflow一样简单吗?

一些看起来很相关的导入语句是:

import theano
import theano.tensor as T
import keras
from keras import backend as K

1 个答案:

答案 0 :(得分:0)

您还需要更改keras后端参数。 documentation中可以通过两种方式完成此操作,一种是更改keras配置文件,另一种是更改脚本顶部的环境变量

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

也许还值得检查是否在其他任何地方使用了theanotheano.tensor,但这应该可以解决问题!