Spaces:
Runtime error
Runtime error
| # -*-coding:utf-8 -*- | |
| """ | |
| Dump Instruction Data for SELF | |
| """ | |
| import json | |
| from ape.instance import LoadFactory | |
| def make_instruct_data(instruction, input_output_list): | |
| data = { | |
| 'instruction': instruction, | |
| 'instances': [] | |
| } | |
| for i, j in input_output_list: | |
| data['instances'].append({'input': i, 'output': j}) | |
| return data | |
| def seed_file(file_name='./ape/data/seed_task.json', n_instances=5): | |
| instruction = { | |
| 'paraphase': '将医学手术名称的术语表述标准化。输入是医学手术的名称，输出是对该手术的名称进行修正、标准化，以供医学专业人员更好地理解', | |
| 'search_intent': '生成医学相关问题的答案。给定一个输入问题，需要根据问题生成相应的输出答案。答案包括临床表现、病因、治疗方法、作用、定义等等，如果有多个问题，返回多问', | |
| 'qa_generation': '训练一个问答系统，给定一些医学文本，能够回答用户提问关于该文本内容的问题。每个输入-输出对是一组文本和对应的问题及答案。输出的形式是以下Json格式{"问题":$问题, "回答":$回答}', | |
| 'entity': '给定药品信息和用途说明，根据用途说明提取出药品的主治功能' | |
| } | |
| with open(file_name, 'w', encoding='UTF8') as f: | |
| for task, instruct in instruction.items(): | |
| sample = LoadFactory[task]()[:n_instances] | |
| data = make_instruct_data(instruct, sample) | |
| f.write(json.dumps(data, ensure_ascii=False) + '\n') | |
| if __name__ == '__main__': | |
| seed_file() | |