mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-04 06:37:45 +00:00
Mark Aron Szulyovszky
797d45d036
* 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>
20 lines
856 B
Python
20 lines
856 B
Python
from training.inference import run_inference_pipeline
|
|
from models.saving import load_models
|
|
|
|
from run_pipeline import run_pipeline
|
|
from config.config import get_dev_config, get_default_ensemble_config, get_lightweight_ensemble_config
|
|
|
|
from typing import Callable
|
|
|
|
|
|
def run_inference(preload_models:bool, get_config:Callable):
|
|
if preload_models:
|
|
all_models_all_assets, data_config, training_config, model_config = load_models(None)
|
|
else:
|
|
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, model_config, all_models_all_assets)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run_inference(preload_models=True, get_config=get_lightweight_ensemble_config) |