Module mls_lib.objects.scalers.scaler

Scaler: Component that performs scaling.

Classes

class Scaler (scaler: IScaler)

Scaler: Component that performs scaling.

Initializes the class instance with a given scaler and column.

Args

scaler : object
The scaler object to be used for scaling the data.
columns : list
A list containing the column name(s) to be scaled.

Returns

None

Expand source code
class Scaler(IScaler):
    """ Scaler: Component that performs scaling. """
    def __init__(self, scaler : IScaler) -> None:
        """
        Initializes the class instance with a given scaler and column.

        Args:
            scaler (object): The scaler object to be used for scaling the data.
            columns (list): A list containing the column name(s) to be scaled.

        Returns:
            None
        """
        super().__init__()
        self.scaler = scaler
        self.columns = []

    def fit_transform(self, data : DataFrame, columns : list):
        """
        Fits the scaler to the data and performs a transform operation on the specified column.

        Parameters:
            data (numpy.ndarray): The input data to be scaled.

        Returns:
            None
        """
        self.columns = columns
        df = data.get_data()
        df[self.columns] = self.scaler.fit_transform(np.array(df[self.columns]).reshape(-1,1))
        data.set_data(df)

    def transform(self, data : DataFrame):
        """
        Transforms the specified column of the input data using the scaler object.

        Parameters:
            data (object): The input data to be transformed.

        Returns:
            None

        """
        df = data.get_data()
        df[self.columns] = self.scaler.transform(np.array(df[self.columns]).reshape(-1,1))
        data.set_data(df)

Ancestors

Subclasses

Methods

def transform(self, data: DataFrame)

Transforms the specified column of the input data using the scaler object.

Parameters

data (object): The input data to be transformed.

Returns

None

Inherited members