Skip to content
Snippets Groups Projects
Commit ebae6a66 authored by Felix Kleinert's avatar Felix Kleinert
Browse files

Merge branch...

Merge branch 'felix_issue287_tech-wrf-datahandler-should-inherit-from-singlestationdatahandler' of ssh://gitlab.version.fz-juelich.de:10022/esde/machine-learning/mlair into felix_issue287_tech-wrf-datahandler-should-inherit-from-singlestationdatahandler
parents f5219ca3 e65376ad
No related branches found
No related tags found
1 merge request!259Draft: Resolve "WRF-Datahandler should inherit from SingleStationDatahandler"
Pipeline #65290 passed
...@@ -486,7 +486,7 @@ class MyLSTMModel(AbstractModelClass): ...@@ -486,7 +486,7 @@ class MyLSTMModel(AbstractModelClass):
super().__init__(input_shape[0], output_shape[0]) super().__init__(input_shape[0], output_shape[0])
# settings # settings
self.dropout_rate = 0.2 self.dropout_rate = 0.25
# apply to model # apply to model
self.set_model() self.set_model()
...@@ -495,8 +495,8 @@ class MyLSTMModel(AbstractModelClass): ...@@ -495,8 +495,8 @@ class MyLSTMModel(AbstractModelClass):
def set_model(self): def set_model(self):
x_input = keras.layers.Input(shape=self._input_shape) 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(16, return_sequences=True, name="First_LSTM", dropout=self.dropout_rate)(x_input)
x_in = keras.layers.LSTM(64, name="Second_LSTM")(x_in) 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) out_main = keras.layers.Dense(self._output_shape, name='Output_Dense')(x_in)
self.model = keras.Model(inputs=x_input, outputs=[out_main]) self.model = keras.Model(inputs=x_input, outputs=[out_main])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment