Multiple Columns Label Encoders

multi_col_label_encoder.py

import pandas as pd
from sklearn.preprocessing import LabelEncoder

class MultiColumnLabelEncoder:
    '''
    modified the scripts of Daria Vasyukova
    '''
    
    def __init__(self, X):
        self.X = X


    def fit(self):
        self.encoders = {}
        for col in self.X.columns:
            self.encoders[col] = LabelEncoder().fit(self.X[col])
        return self


    def transform(self):
        output = self.X.copy()
        for col in self.X.columns:
            output[col] = self.encoders[col].transform(self.X[col])
        return output


    def fit_transform(self):
        return self.fit().transform()


    def inverse_transform(self, transformed_X):
        output = transformed_X.copy()
        for col in self.X.columns:
            output[col] = self.encoders[col].inverse_transform(output[col])
        return output