mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-03 22:27:47 +00:00
13 lines
471 B
Python
13 lines
471 B
Python
|
from sklearn.preprocessing import MinMaxScaler, Normalizer, StandardScaler
|
||
|
|
from typing import Optional, Union
|
||
|
|
from utils.types import ScalerTypes
|
||
|
|
|
||
|
|
def get_scaler(type: ScalerTypes) -> Optional[Union[MinMaxScaler, Normalizer, StandardScaler]]:
|
||
|
|
if type == 'normalize':
|
||
|
|
return Normalizer()
|
||
|
|
elif type == 'minmax':
|
||
|
|
return MinMaxScaler(feature_range= (-1, 1))
|
||
|
|
elif type == 'standardize':
|
||
|
|
return StandardScaler()
|
||
|
|
else:
|
||
|
|
return None
|