feat(Inference): pipeline wired up (#171)

* feat: Basic pipeline extended.

* feat: Added conversion of model list to existing structure (model_name, model_in_time). Fixed loading of previous models and dicts.

* fix: Had an unfinished function.

* fix: Inference wasn't getting model_over_time. Now transformations are not getting it either yet.

Co-authored-by: Daniel Szemerey <szemereydaniel@gmail.com>
This commit is contained in:
Mark Aron Szulyovszky
2022-01-14 10:34:28 +01:00
committed by GitHub
co-authored by Daniel Szemerey
parent 4aefba33ea
commit 797d45d036
9 changed files with 89 additions and 77 deletions
+8
View File
@@ -25,6 +25,14 @@ class Reporting:
self.level: str = level
self.base: list[Reporting.Single_Model] = []
self.metalabeling: list[list[Reporting.Single_Model]] = []
def convert_step_to_tuple(self, step:str)->list[tuple[str, list[Model]]]:
if step == 'base':
return [(x.model_name, x.model_over_time) for x in self.base ]
elif step == 'metalabeling':
return [(x.model_name, x.model_over_time) for sub in self.metalabeling for x in sub]
else:
raise ValueError('Unknown step: {}'.format(step))
class Asset():