From 7c1e95cb5e307e34e9f44bac4b9b122d7a785ebe Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <nikolai.hartmann@gmx.de>
Date: Thu, 29 Nov 2018 11:31:30 +0100
Subject: [PATCH] name target layers in adversarial setup
---
toolkit.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 0403f9b..5ac5fff 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -2107,7 +2107,7 @@ class ClassificationProjectDecorr(ClassificationProject):
)
if (dropout_fraction is not None) and (dropout_fraction > 0):
layers.append(Dropout(rate=dropout_fraction))
- layers.append(Dense(1, activation=self.activation_function_output))
+ layers.append(Dense(1, activation=self.activation_function_output, name="class"))
return self._class_layers
@@ -2120,8 +2120,8 @@ class ClassificationProjectDecorr(ClassificationProject):
self._adv_hidden_layers = []
self._adv_target_layers = []
self._adv_hidden_layers.append(Dense(128, activation="tanh"))
- for binning in self.decorr_binnings:
- layer = Dense(len(binning), activation="softmax")
+ for binning, field_name in zip(self.decorr_binnings, self.target_fields):
+ layer = Dense(len(binning), activation="softmax", name="adv_"+field_name)
self._adv_target_layers.append(layer)
return self._adv_hidden_layers+self._adv_target_layers
--
GitLab