credit.ensemble.crps#

Functions#

calculate_crps_per_channel(→ torch.Tensor)

Calculate CRPS score for each channel.

Module Contents#

credit.ensemble.crps.calculate_crps_per_channel(ensemble_predictions: torch.Tensor, y_true: torch.Tensor) torch.Tensor#

Calculate CRPS score for each channel.

Parameters:
  • ensemble_predictions (torch.Tensor) – Tensor of shape [ensemble_size, 1, channels, 1, height, width] containing ensemble forecasts.

  • y_true (torch.Tensor) – Tensor of shape [1, channels, 1, height, width] with the true values.

Returns:

CRPS scores with shape [channels], where each score is

averaged over the ensemble.

Return type:

torch.Tensor