tabensemb.model.AbstractNN.on_predict_batch_end

method

AbstractNN.on_predict_batch_end(outputs: Any | None, batch: Any, batch_idx: int, dataloader_idx: int) → None

Called in the predict loop after the batch.

Parameters:

• outputs – The outputs of predict_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