Skip to content

Instantly share code, notes, and snippets.

@leigh-johnson

leigh-johnson/keras_lstm.py

Created July 6, 2019 23:28
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 leigh-johnson/8acedbeaf2498212e651840d858abbaf to your computer and use it in GitHub Desktop.
Save leigh-johnson/8acedbeaf2498212e651840d858abbaf to your computer and use it in GitHub Desktop.
Download ZIP
Comparsing TensorFlow vs Keras for an LSTM model
Raw
keras_lstm.py
from keras.layers import LSTM
model = Sequential()
model.add(Embedding(max_features, 256, input_length=maxlen))
model.add(LSTM(output_dim=128, activation='sigmoid', inner_activation='hard_sigmoid'))
model.add(Dropout(0.5)) model.add(Dense(1))
model.add(Activation('sigmoid'))
model.compile(loss='binary_crossentropy', optimizer='rmsprop', metrics=['accuracy'])
model.fit(X_train, Y_train, batch_size=16, nb_epoch=10)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment