From 7853cb06939354d67ce8616347117084c767b09f Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Fri, 30 Nov 2018 08:55:35 +0100
Subject: [PATCH] number of decorr_bins as a parameter
---
toolkit.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index b32e215..a8fe833 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -2025,10 +2025,12 @@ class ClassificationProjectRNN(ClassificationProject):
class ClassificationProjectDecorr(ClassificationProject):
- def __init__(self, *args, **kwargs):
- super(ClassificationProjectDecorr, self).__init__(*args, **kwargs)
+ def _init_from_args(self, name,
+ decorr_bins=10,
+ **kwargs):
+ super(ClassificationProjectDecorr, self)._init_from_args(name, **kwargs)
self.decorr_binnings = []
- self.decorr_bins = 10
+ self.decorr_bins = decorr_bins
self._write_info("project_type", "ClassificationProjectDecorr")
self._class_layers = None
self._adv_hidden_layers = None
--
GitLab