diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py index 930b47a98114482059b355a7968f6044df12413a..9b2058ace77ee7ff4812ff7f4459e85225615553 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])