utils/postprocess.py

import random

def text_mapping(model_output,text_label):
    text_list = text_label[model_output]
    text,sent = random.sample(text_list,1)[0]
    return {'label' : model_output,
            'text' : text,
            'sentiment' : sent}

def text_encoding(output):
    return {'label' : output['label'].encode(encoding='UTF-8', errors='strict'),
            'text' : output['text'].encode(encoding='UTF-8', errors='strict'),
            'sentiment' : output['sentiment'].encode(encoding='UTF-8', errors='strict')}