| 21 | |
| 22 | |
| 23 | class myDataset(Dataset): |
| 24 | def __init__(self, features): |
| 25 | self.features = features |
| 26 | |
| 27 | def __getitem__(self, idx): |
| 28 | return self.features[idx] |
| 29 | |
| 30 | def __len__(self): |
| 31 | return len(self.features) |
| 32 | |
| 33 | |
| 34 | def make_data(args, tokenizer): |