LSTM Model

lstm-model.py

from keras.models import Sequential
from keras.layers.core import Dense, Dropout
from keras.layers import Dense, Embedding, LSTM, SpatialDropout1D

def lstm(optimizer,epochs,batchsize):
  model = Sequential()
  model.add(Embedding(input_dim=vocab_size,
                      output_dim=embedding_dim,
                      input_length=maxlen))
  model.add(SpatialDropout1D(0.4))
  model.add(LSTM(64, activation="tanh"))
  model.add(Dense(3,activation='softmax'))
  model.compile(optimizer=optimizer,
                loss='categorical_crossentropy',
                metrics=['accuracy'])
  history = model.fit(X_train, y_train,
                    epochs=epochs,
                    verbose=0,
                    validation_data=(X_val, y_val),
                    batch_size=batchsize)
  
  return history, model