Module mls_lib.objects.encoders.label_encoder

LabelEncoder: Component that encodes categorical data.

Classes

class LabelEncoder

LabelEncoder: Component that encodes categorical data.

Initializes the class instance with a given encoder.

Args

encoder

The encoder object to be used.

Returns

None

Expand source code

class LabelEncoder(Encoder):
    """ LabelEncoder: Component that encodes categorical data. """
    def __init__(self) -> None:
        super().__init__(LE())
        
    def fit_transform(self, data, columns : list):
        self.columns = columns
        for column in columns:
            data[column] = self.encoder.fit_transform(data[column])
    
    def transform(self, data):
        for column in self.columns:
            data[column] = self.encoder.transform(data[column])

Ancestors

• Encoder
• IEncoder
• Object

Inherited members

• Encoder:
  • copy
  • fit_transform
  • transform