ClassificationData.validate_prediction#

ClassificationData.validate_prediction(batch, model, device, n_classes: Optional[int] = None, eps: float = 0.001)[source]#

Validate the prediction.

Parameters
batcht.Any

Batch as outputed from DataLoader

model: t.Any

Model to run on batch

device: torch.device
n_classesint

Number of classes.

epsfloat , default: 1e-3

Epsilon value to be used in the validation, by default 1e-3