update MyLSTMModel

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])