mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-17 21:08:08 +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
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user