Skip to content

Instantly share code, notes, and snippets.

@sameerg07

sameerg07/keras_model.py

Created August 3, 2018 07:56
Show Gist options
  • Save sameerg07/d9948721568ec583c9d71405271483ad to your computer and use it in GitHub Desktop.
Save sameerg07/d9948721568ec583c9d71405271483ad to your computer and use it in GitHub Desktop.
Download ZIP
Raw
keras_model.py
model = Sequential()
model.add(Conv2D(32, (3, 3), input_shape=input_shape))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(32, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(64, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Flatten())
model.add(Dense(64))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(5))
model.add(Activation('sigmoid'))
model.compile(loss='binary_crossentropy',
optimizer='rmsprop',
metrics=['accuracy'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment