Module mls_lib.feature_engineering.reuse_scaler

Scaler Trainer

Classes

class ReuseScaler (data: DataFrame, scaler: IScaler)

Scaler Trainer

Expand source code

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

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

        scaler.transform(data)

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

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

Ancestors

• FeatureEngineeringStep
• Step

Inherited members

• FeatureEngineeringStep:
  • execute
  • get_output
  • is_finished
  • is_ready