mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-16 12:28:09 +00:00
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:
co-authored by
Daniel Szemerey
parent
4aefba33ea
commit
797d45d036
@@ -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():
|
||||
|
||||
Reference in New Issue
Block a user