Module mls_lib.feature_engineering.scaler_trainer

Scaler Trainer

Classes

class ScalerTrainer (columns: list, data: DataFrame, scaler: IScaler)

Scaler Trainer

Expand source code

class ScalerTrainer(FeatureEngineeringStep):
    """ Scaler Trainer """
    def __init__(self, columns : list, data : DataFrame, scaler : IScaler) -> None:
        super().__init__(data = data)
        self.columns = columns
        self.scaler = scaler

    def execute(self):
        data = self._get_input("data")

        self.scaler.fit_transform(data, self.columns)

        new_data = DataFrame()
        new_data.set_data(data.get_data())

        self._set_output("scaler", self.scaler)

        self._set_output("out", new_data)
        self._finish_execution()

Ancestors

• FeatureEngineeringStep
• Step

Subclasses

• MaxAbsScalerTrainer
• MinMaxScalerTrainer
• RobustScalerTrainer
• StandardScalerTrainer

Inherited members

• FeatureEngineeringStep:
  • execute
  • get_output
  • is_finished
  • is_ready