File size: 548 Bytes
f67c2de |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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) |