mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-21 06:48: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
@@ -6,6 +6,7 @@ import pandas as pd
|
||||
from models.model_map import default_feature_selector_regression, default_feature_selector_classification
|
||||
from models.base import Model
|
||||
from reporting.types import Reporting
|
||||
from typing import Union
|
||||
|
||||
|
||||
def train_meta_labeling_model(
|
||||
@@ -18,7 +19,8 @@ def train_meta_labeling_model(
|
||||
data_config: dict,
|
||||
model_config: dict,
|
||||
training_config: dict,
|
||||
model_suffix: str
|
||||
model_suffix: str,
|
||||
preloaded_models: Union[list[Reporting.Single_Model], None] = None
|
||||
) -> tuple[pd.Series, pd.Series, pd.DataFrame, list[Reporting.Single_Model]]:
|
||||
|
||||
|
||||
@@ -48,7 +50,8 @@ def train_meta_labeling_model(
|
||||
scaler = training_config['scaler'],
|
||||
no_of_classes = 'two',
|
||||
level = 'meta_labeling',
|
||||
print_results = False
|
||||
print_results = False,
|
||||
preloaded_models = preloaded_models
|
||||
)
|
||||
if len(models) > 1:
|
||||
meta_preds = meta_preds.mean(axis = 1)
|
||||
|
||||
Reference in New Issue
Block a user