From 012a05cbb3a3c597781cf6cc2185cefc63399e64 Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Mon, 20 Jan 2020 17:04:57 +0100
Subject: [PATCH] added name to minor output branches
---
src/model_modules/model_class.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/model_modules/model_class.py b/src/model_modules/model_class.py
index 0e31cd66..6b0fe236 100644
--- a/src/model_modules/model_class.py
+++ b/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):
--
GitLab