Lukas issue034 feat plot mse history

src/model_modules/model_class.py

@@ -219,15 +219,15 @@ class MyBranchedModel(AbstractModelClass):
         x_in = keras.layers.Dense(64, name='{}_Dense_64'.format("major"))(x_in)
         x_in = self.activation()(x_in)
         out_minor_1 = keras.layers.Dense(self.window_lead_time, name='{}_Dense'.format("minor_1"))(x_in)
-        out_minor_1 = self.activation()(out_minor_1)
+        out_minor_1 = self.activation(name="minor_1")(out_minor_1)
         x_in = keras.layers.Dense(32, name='{}_Dense_32'.format("major"))(x_in)
         x_in = self.activation()(x_in)
         out_minor_2 = keras.layers.Dense(self.window_lead_time, name='{}_Dense'.format("minor_2"))(x_in)
-        out_minor_2 = self.activation()(out_minor_2)
+        out_minor_2 = self.activation(name="minor_2")(out_minor_2)
         x_in = keras.layers.Dense(16, name='{}_Dense_16'.format("major"))(x_in)
         x_in = self.activation()(x_in)
         x_in = keras.layers.Dense(self.window_lead_time, name='{}_Dense'.format("major"))(x_in)
-        out_main = self.activation()(x_in)
+        out_main = self.activation(name="main")(x_in)
         self.model = keras.Model(inputs=x_input, outputs=[out_minor_1, out_minor_2, out_main])
 
     def set_loss(self):