1. 程式人生 > >Keras 2.0版本執行

Keras 2.0版本執行

Keras 2.0版本執行demo出錯:

d:\program\python3\lib\site-packages\ipykernel_launcher.py:8: UserWarning: Update your `Conv2D` call to the Keras 2 API: `Conv2D(32, (3, 3), activation="relu")`

 

使用Keras時用到了卷積層Convolution2D( )以及Model.fit( ):

x = Convolution2D(8, 3, 3, activation='relu', border_mode='same')(x)
1
Model.fit(x_train, x_train, nb_epoch=10, batch_size=256, shuffle=True, validation_data=(x_test, x_test))
1
遇到了如下錯誤:

UserWarning: Update your `Conv2D` call to the Keras 2 API: `Conv2D(1, (3, 3), padding="same", activation="sigmoid")`
decoded = Convolution2D(1, 3, 3, activation='sigmoid', border_mode='same')(x)

在查看了github上的Keras 2.0發行說明後,發現這是從Keras 1到Keras 2發生的變化.

此處涉及到的有:
Convolution* 層被重新命名 Conv* ;
border_mode - > padding ;
nb_epoch - > epochs;
kernel_size可以設定為一個整數,例如Conv2D(10, 3)相當於Conv2D(10, (3, 3));

因此,

x = Convolution2D(8, 3, 3, activation='relu', border_mode='same')(x)
Model.fit(x_train, x_train, nb_epoch=10, batch_size=256, shuffle=True, validation_data=(x_test, x_test))


改為:

x = Conv2D(8, 3, activation='relu', padding='same')(x)
Model.fit(x_train, x_train, epochs=10, batch_size=256, shuffle=True, validation_data=(x_test, x_test))

其他更多變化參見:Keras 2.0發行說明
原文:https://blog.csdn.net/akadiao/article/details/80405766