(self, month)
| 223 | return data, days |
| 224 | |
| 225 | def random_date_augmentation(self, month): |
| 226 | if self.split == 'train': |
| 227 | return max(0, random.randint(0, self.date_aug_range * 2) - self.date_aug_range + self.monthly_dates[month]) |
| 228 | else: |
| 229 | return self.monthly_dates[month] |
| 230 | |
| 231 | |
| 232 | def collate_fn(batch): |