Module mls_lib.feature_engineering.categoric_boost_encoder_train

CategoricalBoostEncoderTrainer: Component that trains a categorical boost encoder.

Classes

class CatBoostEncoderTrainer (columns, data)

CategoricalBoostEncoderTrainer: Component that trains a categorical boost encoder.

Expand source code

class CatBoostEncoderTrainer(EncoderTrainer):
    """ CategoricalBoostEncoderTrainer: Component that trains a categorical boost encoder. """
    def __init__(self, columns, data):
        super().__init__(
            columns = columns,
            data = data,
            encoder = CategoricalBoostEncoder()
        )

Ancestors

• EncoderTrainer
• FeatureEngineeringStep
• Step

Inherited members

• EncoderTrainer:
  • execute
  • get_output
  • is_finished
  • is_ready