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')}