class Decoder:
    def __init__(self):
        import pickle
        with open('pipeline/preprocessing/encoder_toxicity.pkl', 'rb') as v:
            self.__encoder_toxicity = pickle.load(v)
        with open('pipeline/preprocessing/encoder_emotion.pkl', 'rb') as v:
            self.__encoder_emotion = pickle.load(v)
    
    # Decoding one-hot encoded labels
    def toxicity(self,pred):
        return self.__encoder_toxicity.inverse_transform(pred)
    
    def emotion(self,pred):
        return self.__encoder_emotion.inverse_transform(pred)