mirror of
https://github.com/webclinic017/drift.git
synced 2026-07-28 11:17:47 +00:00
Mark Aron Szulyovszky
1856fcad22
* feat(Transformations): added Transformations abstraction & handling in walk_forward_train() & inference() * fix(WalkForward): use Dataframes to call Transformation.fit_transform() * feat(WalkForward): restored option for models to recieve unscaled data * fix(Transformations): output DataFrame as expected * fix(Tests): missing new property
32 lines
744 B
Python
32 lines
744 B
Python
from __future__ import annotations
|
|
from typing import Literal, Optional, Union
|
|
from abc import ABC, abstractmethod
|
|
import pandas as pd
|
|
|
|
class Transformation(ABC):
|
|
|
|
@abstractmethod
|
|
def fit(self, X: pd.DataFrame, y: Optional[pd.Series]) -> None:
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def fit_transform(self, X: pd.DataFrame, y: Optional[pd.Series] = None) -> pd.DataFrame:
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def transform(self, X: pd.DataFrame) -> pd.DataFrame:
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def clone(self) -> Transformation:
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
def get_name(self) -> str:
|
|
raise NotImplementedError
|
|
|
|
|
|
|
|
|
|
|