(self)
| 98 | ] |
| 99 | |
| 100 | def test_dataloader(self): |
| 101 | if len(self.eval_splits) == 1: |
| 102 | return self.plugin.prepare_dataloader(self.dataset["test"], batch_size=self.eval_batch_size) |
| 103 | elif len(self.eval_splits) > 1: |
| 104 | return [ |
| 105 | self.plugin.prepare_dataloader(self.dataset[x], batch_size=self.eval_batch_size) |
| 106 | for x in self.eval_splits |
| 107 | ] |
| 108 | |
| 109 | def convert_to_features(self, example_batch): |
| 110 | # Either encode single sentence or sentence pairs |
no test coverage detected