Skip to content

Instantly share code, notes, and snippets.

@steermomo
Created April 15, 2018 12:33
Show Gist options
  • Save steermomo/7efc33390458004987fa29898efd443e to your computer and use it in GitHub Desktop.
Save steermomo/7efc33390458004987fa29898efd443e to your computer and use it in GitHub Desktop.
Keras TypeError: max_pool3d() got an unexpected keyword argument 'data_format'
def classifier(input_shape, kernel_size, pool_size):
model = Sequential()
model.add(Convolution3D(16, kernel_size=kernel_size,
padding='valid',
input_shape=input_shape))
model.add(Activation('relu'))
model.add(MaxPooling3D(pool_size=pool_size))
model.add(Convolution3D(32, kernel_size=kernel_size))
model.add(Activation('relu'))
model.add(MaxPooling3D(pool_size=pool_size))
model.add(Convolution3D(64, kernel_size=kernel_size))
model.add(Activation('relu'))
model.add(MaxPooling3D(pool_size=pool_size))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(512))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(128))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(2))
model.add(Activation('softmax'))
return model
def train_classifier(input_shape):
model = classifier(input_shape, (3, 3, 3), (2, 2, 2))
model.compile(loss='categorical_crossentropy',
optimizer='adadelta',
metrics=['accuracy'])
train_x = glob('working_space/traindata_*_Xtrain.p')
train_y = glob('working_space/traindata_*_Ytrain.p')
file_tup = list(zip(train_x, train_y))
np.random.shuffle(file_tup)
train_length = int(len(file_tup) * 0.8)
train_x_path = [x[0] for x in file_tup[:train_length]]
train_y_path = [x[1] for x in file_tup[:train_length]]
val_x_path = [x[0] for x in file_tup[train_length:]]
val_y_path = [x[1] for x in file_tup[train_length:]]
# model.train_on_batch(trainX, trainY, sample_weight=None)
model.fit_generator(data_generator(train_x_path, train_y_path),
validation_data=data_generator(val_x_path, val_y_path), steps_per_epoch=108, validation_steps=10)
model.save('CNN_model.h5')
train_classifier((36,36,36, 1))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment