2016-09-29 55 views

回答

0

的官方文檔是在這裏:https://keras.io/backend/

基本上,編輯您$HOME/.keras/keras.json(Linux)或%USERPROFILE%\.keras\keras.json(窗口)的配置文件。

用途: { "image_data_format": "channels_last", "epsilon": 1e-07, "floatx": "float32", "backend": "theano" }

(請注意 「後臺」 設置爲 「theano」。)

當然變化所有您Keras項目中使用Theano這必然。

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

(我測試了在Windows 10:

如果你只是想改變1個項目,你可以在導入keras之前設置KERAS_BACKEND環境變量,無論是從命令行或代碼Python 3.5安裝了Theano和TensorFlow(刪除它,它使用TensorFlow,包含它,它將使用Theano))。

很高興在您的Python源代碼中包含這個依賴關係,因爲它明確包含在源代碼管理中。由於Keras所使用的底層ML庫不是100%抽象的(有許多微小的差異可以滲透),讓代碼表明它需要一個或另一個可能是一個好主意。

我希望幫助,

羅伯特