(self, cal_lead=False, cal_oracle=False)
| 137 | init_logger(args.log_file) |
| 138 | |
| 139 | def baseline(self, cal_lead=False, cal_oracle=False): |
| 140 | test_iter = data_loader.DataLoader(self.args, data_loader.load_dataset(self.args, 'test', shuffle=False), |
| 141 | self.args.batch_size, self.device, shuffle=False, is_test=True) |
| 142 | |
| 143 | trainer = build_trainer(self.args, self.device_id, None, None) |
| 144 | |
| 145 | if cal_lead: |
| 146 | trainer.test(test_iter, 0, cal_lead=True) |
| 147 | elif cal_oracle: |
| 148 | trainer.test(test_iter, 0, cal_oracle=True) |
| 149 | |
| 150 | def train_iter(self): |
| 151 | return data_loader.DataLoader(self.args, data_loader.load_dataset(self.args, 'train', shuffle=True), |
no test coverage detected