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