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
+7 -3
View File
@@ -5,6 +5,7 @@ from training.primary_model import train_primary_model
from training.meta_labeling import train_meta_labeling_model
from reporting.types import Reporting
from typing import Union
def primary_step(
@@ -14,7 +15,8 @@ def primary_step(
asset:list,
target_returns:pd.Series,
configs: dict,
reporting: Reporting
reporting: Reporting,
preloaded_training_step: Union[Reporting.Training_Step, None] = None
) -> tuple[Reporting.Training_Step, pd.DataFrame]:
training_step = Reporting.Training_Step(level='primary')
model_config, training_config, data_config = itemgetter('model_config', 'training_config', 'data_config')(configs)
@@ -34,7 +36,8 @@ def primary_step(
scaler = training_config['scaler'],
no_of_classes = data_config['no_of_classes'],
level = 'primary',
print_results= True
print_results= True,
preloaded_models = preloaded_training_step.convert_step_to_tuple('base') if preloaded_training_step is not None else None
)
training_step.base = all_models_for_single_asset
@@ -53,7 +56,8 @@ def primary_step(
data_config= data_config,
model_config= model_config,
training_config= training_config,
model_suffix = 'meta'
model_suffix = 'meta',
preloaded_models =preloaded_training_step.convert_step_to_tuple('metalabeling') if preloaded_training_step is not None else None
)
current_result[model_name] = primary_meta_result
current_predictions[model_name] = primary_meta_preds