Keras 后端配置

2021-11-01 09:43 更新

本章將詳細(xì)解釋 Keras 后端實(shí)現(xiàn) TensorFlowTheano。下面,讓我們一一介紹每個(gè)的實(shí)現(xiàn)。

TensorFlow

TensorFlow 是谷歌開發(fā)的用于數(shù)值計(jì)算任務(wù)的開源機(jī)器學(xué)習(xí)庫。

Keras 是一個(gè)建立在 TensorFlow 或 Theano 之上的高級(jí) API。

上一篇內(nèi)容,我們已經(jīng)介紹過 TensorFlow 的安裝,如果還沒有安裝,使用下面命令進(jìn)行安裝即可:

pip install TensorFlow

一旦我們執(zhí)行 Keras,就可以看到配置文件位于你的主目錄里面,下面點(diǎn)開.keras/keras.json文件。

keras.json

{
   "image_data_format":"channels_last",
   "epsilon":1e-07,
   "floatx":"float32",
   "backend":"tensorflow"
}

這里面:

  • image_data_format 表示數(shù)據(jù)格式。
  • epsilon 表示數(shù)字常量,用于避免 DivideByZero 錯(cuò)誤。
  • floatx 表示默認(rèn)數(shù)據(jù)類型 float32。此外,還可以用 set_floatx() 方法將其更改為 float16 或者 float64
  • backend 表示后端配置。

假設(shè),如果還沒有創(chuàng)建文件,則可以移動(dòng)到該位置并用以下步驟創(chuàng)建:

>cd home
>mkdir .keras
>vi keras.json

注意:指定 .keras 作為其文件夾名稱,并在 keras.json 文件中添加上述的配置。這樣,我們才可以執(zhí)行一些預(yù)定義的操作來了解后端功能。

Theano

Theano 是一個(gè)開源深度學(xué)習(xí)庫,可以讓你有效地評(píng)估多維數(shù)組。我們可以使用下列命令安裝:

pip install theano

默認(rèn)情況下,keras 使用 TensorFlow 后端。如果想要把后端配置從 TensorFlow 改為 Theano,只需要更改 keras.json 文件中的 backend = theano 即可。它的描述具體如下:

keras.json

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

接下來,保存你的文件,重新啟動(dòng)終端并啟動(dòng) Keras,你的后端就會(huì)被更改。

>>> import keras as k
using theano backend.
以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)