Skip to content

Instantly share code, notes, and snippets.

@curiousily
Created April 23, 2020 19:51
Show Gist options
  • Save curiousily/70edaff421c81f8c8261648edaebe60b to your computer and use it in GitHub Desktop.
Save curiousily/70edaff421c81f8c8261648edaebe60b to your computer and use it in GitHub Desktop.
model = keras.Sequential()
model.add(
keras.layers.Bidirectional(
keras.layers.LSTM(
units=128,
input_shape=[X_train.shape[1], X_train.shape[2]]
)
)
)
model.add(keras.layers.Dropout(rate=0.5))
model.add(keras.layers.Dense(units=128, activation='relu'))
model.add(keras.layers.Dense(y_train.shape[1], activation='softmax'))
model.compile(
loss='categorical_crossentropy',
optimizer='adam',
metrics=['acc']
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment