diff --git a/toolkit.py b/toolkit.py index b32e2152f62020da0fe6fd12228779fea1ee41a1..a8fe83334a688a1f11d5d4f44332670b60189464 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