Spaces:

Gae8J
/

gaepago_model

Paused

App Files Files Community

gaepago_model / utils /postprocess.py

khan123

Update utils/postprocess.py

91419c9 over 3 years ago

Raw History Blame Contribute Delete

516 Bytes

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

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