MNISTData.validate_prediction#
- MNISTData.validate_prediction(batch, model, device)[source]#
Validate the prediction.
- Parameters
- batcht.Any
Batch from DataLoader
- modelt.Any
- devicetorch.Device
- Raises
- ValidationError
If predictions format is invalid (depends on validate_infered_batch_predictions implementations)
- DeepchecksNotImplementedError
If infer_on_batch not implemented