project Refactor, CNN Classifier Basics

2020-02-19 21:11:42 +01:00
parent 8424251ca0
commit 78f0df8a2a
16 changed files with 622 additions and 560 deletions
--- a/lib/utils/model_io.py
+++ b/lib/utils/model_io.py
@ -14,22 +14,10 @@ class ModelParameters(Namespace):
        tanh=nn.Tanh
    )

-    @property
-    def model_param(self):
-        return self._model_param
-
-    @property
-    def train_param(self):
-        return self._train_param
-
-    @property
-    def data_param(self):
-        return self._data_param
-
    def __init__(self, model_param, train_param, data_param):
-        self._model_param = model_param
-        self._train_param = train_param
-        self._data_param = data_param
+        self.model_param = model_param
+        self.train_param = train_param
+        self.data_param = data_param
        kwargs = vars(model_param)
        kwargs.update(vars(train_param))
        kwargs.update(vars(data_param))