From e65376ad3d938eedf4c90613174779bad74e3105 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Sun, 11 Apr 2021 08:56:34 +0200
Subject: [PATCH] update MyLSTMModel
---
mlair/model_modules/model_class.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py
index 930b47a9..9b2058ac 100644
--- a/mlair/model_modules/model_class.py
+++ b/mlair/model_modules/model_class.py
@@ -486,7 +486,7 @@ class MyLSTMModel(AbstractModelClass):
super().__init__(input_shape[0], output_shape[0])
# settings
- self.dropout_rate = 0.2
+ self.dropout_rate = 0.25
# apply to model
self.set_model()
@@ -495,8 +495,8 @@ class MyLSTMModel(AbstractModelClass):
def set_model(self):
x_input = keras.layers.Input(shape=self._input_shape)
- x_in = keras.layers.LSTM(32, return_sequences=True, name="First_LSTM")(x_input)
- x_in = keras.layers.LSTM(64, name="Second_LSTM")(x_in)
+ x_in = keras.layers.LSTM(16, return_sequences=True, name="First_LSTM", dropout=self.dropout_rate)(x_input)
+ x_in = keras.layers.LSTM(32, name="Second_LSTM", dropout=self.dropout_rate)(x_in)
out_main = keras.layers.Dense(self._output_shape, name='Output_Dense')(x_in)
self.model = keras.Model(inputs=x_input, outputs=[out_main])
--
GitLab