aggregate_metrics#

aggregate_metrics(outputs: List[Dict[str, Union[float, torch.Tensor]]], exclude_keys: Container[str] = {'_', 'a', 'c', 'f', 'm', 'n', 'o', 't'}) Dict[str, float][source]#

Aggregate the output of validate_step at the end of the epoch.

Parameters
  • outputs (List[Dict[str, Union[float, torch.Tensor]]]) – A list of outputs of Learner.validate_step().

  • exclude_keys (Container[str]) – Keys to ignore. These will not be aggregated and will not be included in the output. Defaults to {‘conf_mat’}.

Returns

Dict with aggregated values.

Return type

Dict[str, float]