updating char based model

jshtaway · Oct 26, 2018 · deecfbb · deecfbb
1 parent 868eeb9
commit deecfbb
Show file tree

Hide file tree

Showing 2 changed files with 433 additions and 3 deletions.
diff --git a/character_based/info_{now.strftime('%Y-%m-%d_%H-%M')}.txt b/character_based/info_{now.strftime('%Y-%m-%d_%H-%M')}.txt
@@ -1,5 +1,8 @@
-{'seq_length': 100}model = Sequential()
-model.add(LSTM(256, input_shape=(X.shape[1], X.shape[2])))
+{'seq_length': 100}
+model = Sequential()
+model.add(LSTM(400, input_shape=(X.shape[1], X.shape[2]), return_sequences=True))
+model.add(Dropout(0.2))
+model.add(LSTM(200))
 model.add(Dropout(0.2))
 model.add(Dense(y.shape[1], activation='softmax'))
-model.compile(loss='categorical_crossentropy', optimizer='adam')
+model.compile(loss='categorical_crossentropy', optimizer='adam')