mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-23 07:48:09 +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:
@@ -26,8 +26,10 @@ class RawConfig(BaseModel):
|
||||
other_features: list[str]
|
||||
exogenous_features: list[str]
|
||||
event_filter: Literal["none", "cusum_vol", "cusum_fixed"]
|
||||
remove_overlapping_events: bool
|
||||
labeling: Literal["two_class", "three_class_balanced", "three_class_imbalanced"]
|
||||
forecasting_horizon: int
|
||||
transaction_costs: float
|
||||
save_models: bool
|
||||
ensembling_method: Literal["voting_soft", "stacking"]
|
||||
|
||||
@@ -49,8 +51,10 @@ class Config:
|
||||
other_features: list[tuple[str, FeatureExtractor, list[int]]]
|
||||
exogenous_features: list[tuple[str, FeatureExtractor, list[int]]]
|
||||
event_filter: EventFilter
|
||||
remove_overlapping_events: bool
|
||||
labeling: EventLabeller
|
||||
forecasting_horizon: int
|
||||
transaction_costs: float
|
||||
no_of_classes: Literal["two", "three-balanced", "three-imbalanced"]
|
||||
save_models: bool
|
||||
|
||||
|
||||
Reference in New Issue
Block a user