tabensemb.model.AbstractNN.on_test_batch_end

method

AbstractNN.on_test_batch_end(outputs: Tensor | Dict[str, Any] | None, batch: Any, batch_idx: int, dataloader_idx: int) → None

Called in the test loop after the batch.

Parameters:

• outputs – The outputs of test_step_end(test_step(x))

• batch – The batched data as it is returned by the test DataLoader.

• batch_idx – the index of the batch

• dataloader_idx – the index of the dataloader