drift/utils/scaler.py

from sklearn.preprocessing import MinMaxScaler, Normalizer, StandardScaler
from transformations.sklearn import SKLearnTransformation
from utils.types import ScalerTypes

def get_scaler(type: ScalerTypes) -> SKLearnTransformation:
    if type == 'normalize':
        return SKLearnTransformation(Normalizer())
    elif type == 'minmax':
        return SKLearnTransformation(MinMaxScaler(feature_range= (-1, 1)))
    elif type == 'standardize':
        return SKLearnTransformation(StandardScaler())
    else:
        raise Exception("Scaler type not supported")