File size: 1,396 Bytes
256a159 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import json
from datasets import Dataset
from opencompass.registry import LOAD_DATASET, TEXT_POSTPROCESSORS
from .base import BaseDataset
@LOAD_DATASET.register_module()
class DRCDDataset(BaseDataset):
@staticmethod
def load(path: str):
with open(path, 'r', encoding='utf-8') as f:
data = json.load(f)
# 将原始数据转换为所需的格式
rows = []
for index, paragraphs in enumerate(data['data']):
for paragraph in paragraphs['paragraphs']:
context = paragraph['context']
for question in paragraph['qas']:
answers = question['answers']
unique_answers = list(set([a['text'] for a in answers]))
rows.append({
'context': context,
'question': question['question'],
'answers': unique_answers
})
# 创建 Dataset
dataset = Dataset.from_dict({
'context': [row['context'] for row in rows],
'question': [row['question'] for row in rows],
'answers': [row['answers'] for row in rows]
})
return dataset
@TEXT_POSTPROCESSORS.register_module('drcd')
def drcd_postprocess(text: str) -> str:
if '答案是' in text:
text = text.split('答案是')[1]
return text
|