model = Sequential()
K.set_image_dim_ordering('th')
model.add(Convolution2D(30, 5, 5, border_mode= 'valid' , input_shape=(1, 10, 10),activation= 'relu' ))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Convolution2D(15, 3, 3, activation= 'relu' ))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.2))
model.add(Flatten())
model.add(Dense(128, activation= 'relu' ))
model.add(Dense(50, activation= 'relu' ))
model.add(Dense(10, activation= 'softmax' ))
# Compile model
model.compile(loss= 'categorical_crossentropy' , optimizer= 'adam' , metrics=[ 'accuracy' ])
Run Code Online (Sandbox Code Playgroud)
set_image_dim_ordering()当我使用from时出现错误keras.backend
这是错误报告:AttributeError: module 'keras.backend' has no attribute 'set_image_dim_ordering'
我的进口声明
import numpy as np # linear algebra
import pandas as pd # data processing, CSV file I/O …Run Code Online (Sandbox Code Playgroud)