Module mls_lib.feature_engineering.reuse_encoder

Encoder Trainer

Classes

class EncoderTrainer (data: DataFrame, encoder: IEncoder)

Encoder Trainer

Expand source code

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

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

        df = data.get_data()

        encoder.transform(df)

        data.set_data(df)

        self._set_output("encoder", encoder)

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

Ancestors

• FeatureEngineeringStep
• Step

Inherited members

• FeatureEngineeringStep:
  • execute
  • get_output
  • is_finished
  • is_ready