Skip to content

Instantly share code, notes, and snippets.

@vivekpadia70

vivekpadia70/imdb_lstm.py

Created April 8, 2020 11:52
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 vivekpadia70/17e22a524cf287445b5910ad3a7e057e to your computer and use it in GitHub Desktop.
Save vivekpadia70/17e22a524cf287445b5910ad3a7e057e to your computer and use it in GitHub Desktop.
Download ZIP
Raw
imdb_lstm.py
model = tf.keras.Sequential()
model.add(tf.keras.layers.Embedding(13000, 16, input_length=max_len_text))
model.add(tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(64, dropout=0.2)))
model.add(tf.keras.layers.Dense(1, activation='sigmoid'))
model.summary()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment