15 lines
355 B
Python
15 lines
355 B
Python
from __future__ import absolute_import
|
|
from keras.layers.core import Dense, Activation, Flatten
|
|
|
|
|
|
def layers(input_shape):
|
|
return [
|
|
Dense(128, input_shape=input_shape),
|
|
Activation('relu'),
|
|
Dense(128, input_shape=input_shape),
|
|
Activation('relu'),
|
|
Flatten(),
|
|
Dense(128),
|
|
Activation('relu'),
|
|
]
|