Module mls_lib.feature_engineering.encoder_trainer

Encoder Trainer

Classes

class EncoderTrainer (columns: list, data: DataFrame, encoder: IEncoder)

Encoder Trainer

Expand source code

class EncoderTrainer(FeatureEngineeringStep):
    """ Encoder Trainer """
    def __init__(self, columns : list, data : DataFrame, encoder : IEncoder) -> None:
        super().__init__(data = data)
        self.columns = columns
        self.encoder = encoder

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

        self.encoder.fit_transform(df, self.columns)

        data.set_data(df)

        self._set_output("encoder", self.encoder)

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

Ancestors

• FeatureEngineeringStep
• Step

Subclasses

• CatBoostEncoderTrainer
• LabelEncoderTrainer
• OneHotEncoderTrainer

Inherited members

• FeatureEngineeringStep:
  • execute
  • get_output
  • is_finished
  • is_ready