Update LSTM network.

1d5e2c5b · Matt Harvey · 47fc0a4d · 1d5e2c5b
Commit 1d5e2c5b authored 7 years ago by Matt Harvey
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 2 deletions
+2 -2
--- a/models.py
+++ b/models.py
@@ -71,10 +71,10 @@ class ResearchModels():
        our CNN to this model predomenently."""
        # Model.
        model = Sequential()
-        model.add(LSTM(256, return_sequences=True, input_shape=self.input_shape,
+        model.add(LSTM(4096, return_sequences=True, input_shape=self.input_shape,
                       dropout_W=0.5, dropout_U=0.5))
        model.add(Flatten())
-        model.add(Dense(512, activation='relu'))
+        model.add(Dense(1024, activation='relu'))
        model.add(Dropout(0.5))
        model.add(Dense(self.nb_classes, activation='softmax'))