mirror of
https://github.com/webclinic017/drift.git
synced 2026-07-28 11:17:47 +00:00
Mark Aron Szulyovszky
13 lines
396 B
Python
13 lines
396 B
Python
from typing import Protocol, Callable, Union
|
|
import pandas as pd
|
|
|
|
class SKLearnModel(Protocol):
|
|
def fit(self, X, y, sample_weight=None): ...
|
|
def predict(self, X): ...
|
|
def score(self, X, y, sample_weight=None): ...
|
|
def set_params(self, **params): ...
|
|
|
|
|
|
Period = int
|
|
IsLogReturn = bool
|
|
FeatureExtractor = Callable[[pd.DataFrame, Period, IsLogReturn], Union[pd.DataFrame, pd.Series]] |