Module mls_lib.feature_engineering.one_hot_encoder_trainer

OneHotEncoderTrainer: Component that trains a one hot encoder.

Classes

class OneHotEncoderTrainer (columns, data)

OneHotEncoderTrainer: Component that trains a one hot encoder.

Expand source code

class OneHotEncoderTrainer(EncoderTrainer):
    """ OneHotEncoderTrainer: Component that trains a one hot encoder. """
    def __init__(self, columns, data):
        super().__init__(
            columns = columns,
            data = data,
            encoder = OneHotEncoder()
        )

Ancestors

• EncoderTrainer
• FeatureEngineeringStep
• Step

Inherited members

• EncoderTrainer:
  • execute
  • get_output
  • is_finished
  • is_ready