gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-18 05:18:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files
run_inference.py
+4 -4
View File
@@ -9,12 +9,12 @@ from typing import Callable
def run_inference(preload_models:bool, get_config:Callable):
if preload_models:
all_models_all_assets, data_config, training_config = load_models(None)
all_models_all_assets, data_config, training_config, model_config = load_models(None)
else:
all_models_all_assets, data_config, training_config, _, _, _ = run_pipeline(project_name='price-prediction', with_wandb = False, sweep = False, get_config=get_config)
all_models_all_assets, data_config, training_config, model_config, _, _, _ = run_pipeline(project_name='price-prediction', with_wandb = False, sweep = False, get_config=get_config)
run_inference_pipeline(data_config, training_config, all_models_all_assets)
run_inference_pipeline(data_config, training_config, model_config, all_models_all_assets)
if __name__ == '__main__':
run_inference(preload_models=False, get_config=get_lightweight_ensemble_config)
run_inference(preload_models=True, get_config=get_lightweight_ensemble_config)