mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-26 09:18:05 +00:00
feat(Labeling): purge overlapping events, sort dataframe at loading time (#226)
* feat(Labeling): purge overlapping events, sort dataframe at loading time * fix(Linter): ran * refactor(Labeling): moved purge_overlapping_events one abstraction level higher * fix(Data): renamed class * fix(Data): corrected parameter name * fix(Config): parameters * fix(Data): fixed path * fix(Data): uncommented required code * feat(EventFilters): use vol based CUSUM * fix(Config): only retrain every 2000 samples * fix(Config): filter out even more events * fix(Inference): added remove_overlapping_events * refactor(Types): simplified type hierarchy
This commit is contained in:
@@ -2,7 +2,7 @@ import pandas as pd
|
||||
|
||||
from transformations.base import Transformation
|
||||
|
||||
from .types import DirectionalTrainingOutcome, TrainingOutcome
|
||||
from .types import TrainingOutcome
|
||||
from training.train_model import train_model
|
||||
from training.walk_forward import walk_forward_process_transformations
|
||||
|
||||
@@ -19,8 +19,8 @@ def train_directional_model(
|
||||
model: Model,
|
||||
transformations: list[Transformation],
|
||||
from_index: Optional[pd.Timestamp],
|
||||
preloaded_training_step: Optional[DirectionalTrainingOutcome] = None,
|
||||
) -> DirectionalTrainingOutcome:
|
||||
preloaded_training_step: Optional[TrainingOutcome] = None,
|
||||
) -> TrainingOutcome:
|
||||
|
||||
if preloaded_training_step is None:
|
||||
print("Preprocess transformations")
|
||||
@@ -49,11 +49,13 @@ def train_directional_model(
|
||||
level="primary",
|
||||
output_stats=config.mode == "training",
|
||||
transformations_over_time=transformations_over_time,
|
||||
model_over_time=preloaded_training_step.training.model_over_time
|
||||
model_over_time=preloaded_training_step.model_over_time
|
||||
if preloaded_training_step
|
||||
else None,
|
||||
)
|
||||
if config.mode == "training":
|
||||
print(training_outcome.stats)
|
||||
|
||||
return DirectionalTrainingOutcome(training_outcome, transformations_over_time)
|
||||
return TrainingOutcome(
|
||||
**vars(training_outcome), transformations=transformations_over_time
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user