def model(learning_rate, input_shape):
mdl = Model(learning_rate)
x = mdl.add(Convolution(Input(input_shape), (2, 10), 'relu', 'conv_1'))
x = mdl.add(Convolution(x, (2, 10), 'relu', 'conv_2'))
x = mdl.add(Flatten(x, name='flatten_1'))
mdl.add(Dense(x, 1, 'sigmoid', 'out_1'))
return mdl
def model(learning_rate, input_shape):
mdl = Model(learning_rate)
x = mdl.add(Dense(Input(input_shape), 4, 'relu', 'dense_1'))
x = mdl.add(Dense(x, 2, 'relu', 'dense_2'))
mdl.add(Dense(x, 1, 'sigmoid', 'out_1'))
return mdl
