(self, batch)
| 202 | return MultiChoiceTaskDataset(join(self.config.path, relative_path), self.config) |
| 203 | |
| 204 | def predict_single_batch(self, batch) -> List[int]: |
| 205 | log_probs = self.model.cond_log_prob(batch) |
| 206 | return [np.argmax(log_probs_single).item() for log_probs_single in log_probs] |
| 207 | |
| 208 | |
| 209 | class LanguageModelTask(BaseTask, ABC): |
nothing calls this directly
no test coverage detected