Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/model.py

Created March 25, 2018 18:03
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save NMZivkovic/5c09347f7b09ffd0997b6d08375bc88d to your computer and use it in GitHub Desktop.
Save NMZivkovic/5c09347f7b09ffd0997b6d08375bc88d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model.py
# Define network architecture and compile
model = Sequential()
model.add(Embedding(num_words, 50, input_length=200))
model.add(Dropout(0.2))
model.add(LSTM(100, dropout=0.2, recurrent_dropout=0.2))
model.add(Dense(250, activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(1, activation='sigmoid'))
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment