gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-06 23:57:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(Models): added simple Feed forward layer

Browse Source
This commit is contained in:
Mark Aron Szulyovszky
2021-11-11 18:42:23 +01:00
parent 5f8efc97fa
commit 67d830ca09
2 changed files with 99 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model_lstm.py
+3 -3
View File
@@ -74,10 +74,10 @@ print(batch_targets)
# %%
model = keras.Sequential()
model.add(keras.layers.LSTM(units = 32, return_sequences = True, activation = 'relu', input_shape=(batch_inputs.shape[1], batch_inputs.shape[2])))
# model.add(keras.layers.Dropout(0.4))
# model.add(keras.layers.Dense(units = 10, activation = 'relu'))
model.add(keras.layers.Dropout(0.4))
model.add(keras.layers.Dense(units = 1, activation = 'relu'))
model.add(keras.layers.Dense(units = 10, activation = 'relu'))
model.add(keras.layers.Dropout(0.4))
model.add(keras.layers.Dense(units = 1))
optimizer = keras.optimizers.Adam(learning_rate=learning_rate, clipnorm=1.0)
model.compile(optimizer=optimizer, loss="mean_squared_error")