(self, batch: Any, batch_idx: int, dataloader_idx: int = None)
| 284 | return main_test_stats |
| 285 | |
| 286 | def predict_step(self, batch: Any, batch_idx: int, dataloader_idx: int = None): |
| 287 | return self._evaluation_step(batch, batch_idx) |
| 288 | |
| 289 | def on_predict_epoch_end(self, results: List[Any]) -> Dict[int, Dict[str, Dict[str, np.ndarray]]]: |
| 290 | return self.aggregate_predictions(results) |
nothing calls this directly
no test coverage detected