Skip to content

Instantly share code, notes, and snippets.

@ikamensh
Created June 23, 2017 10:09
Show Gist options
  • Save ikamensh/b2b2ff205e482b801dd0d284c86a2dd1 to your computer and use it in GitHub Desktop.
Save ikamensh/b2b2ff205e482b801dd0d284c86a2dd1 to your computer and use it in GitHub Desktop.
model used for binary classification. For bug report of Keras.save / Keras.load
from keras.models import Sequential
from keras.layers import Dense, Dropout, LeakyReLU
from keras.optimizers import adam
ad = adam()
size_factor=128
predictor=Sequential()
predictor.add(Dense(4*size_factor, input_dim=10))
predictor.add(LeakyReLU())
predictor.add(Dense(4*size_factor))
predictor.add(LeakyReLU())
predictor.add(Dense(4*size_factor))
predictor.add(LeakyReLU())
predictor.add(Dropout(0.3))
predictor.add(Dense(2*size_factor))
predictor.add(LeakyReLU())
predictor.add(Dropout(0.3))
predictor.add(Dense(size_factor))
predictor.add(LeakyReLU())
predictor.add(Dropout(0.3))
predictor.add(Dense(1, activation='sigmoid'))
predictor.compile(adam(),loss='binary_crossentropy')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment