| import json | |
| from datasets import Dataset, DatasetDict | |
| from .base import BaseDataset | |
| class CrowspairsDataset_CN(BaseDataset): | |
| def load(path): | |
| data = [] | |
| with open(path, 'r') as f: | |
| for line in f: | |
| item = json.loads(line) | |
| data.append(item) | |
| def preprocess(example): | |
| example['label'] = 'A' | |
| return example | |
| dataset = Dataset.from_list(data).map(preprocess) | |
| return DatasetDict({'test': dataset}) | |