Adds support for multi-output models.

Implements #154 and fixes #353.
autonomio · Jul 24, 2019 · 2360cc5 · 2360cc5
1 parent 42951e9
commit 2360cc5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/talos/scan/scan_finish.py b/talos/scan/scan_finish.py
@@ -43,7 +43,10 @@ def scan_finish(self):
     except AttributeError:
         out['x_shape'] = 'list'
 
-    out['y_shape'] = self.y.shape
+    try:
+        out['y_shape'] = self.y.shape
+    except AttributeError:
+        out['y_shape'] = 'list'
 
     # final cleanup
     keys = list(self.__dict__.keys())

diff --git a/talos/scan/scan_prepare.py b/talos/scan/scan_prepare.py
@@ -44,19 +44,11 @@ def scan_prepare(self):
     self.saved_models = []
     self.saved_weights = []
 
-    # create the data asset
-    self.y_max = self.y.max()
-
     # handle validation split
     from ..utils.validation_split import validation_split
     self = validation_split(self)
 
     # set data and len
-    self.shape = [self.x.shape, self.y.shape]
     self._data_len = len(self.x)
 
-    # infer prediction type
-    from ..utils.detector import prediction_type
-    self = prediction_type(self)
-
     return self