incerto.utils.evaluate

Contents

incerto.utils.evaluate#

incerto.utils.evaluate(model, data_loader, criterion, device, desc='Evaluating')[source]#

Evaluate model on dataset.

Parameters:
  • model (Module) – Model to evaluate

  • data_loader (DataLoader) – Data loader

  • criterion (Module) – Loss function

  • device (device) – Device to evaluate on

  • desc (str) – Description for progress bar

Return type:

dict

Returns:

Dictionary with ‘loss’, ‘accuracy’, ‘logits’, ‘targets’