fix: mismatched arguments when calling calc_quantile_loss;

WenjieDu · Feb 19, 2025 · 5f18d30 · 5f18d30
1 parent f5d4e24
commit 5f18d30
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pypots/utils/metrics/error.py b/pypots/utils/metrics/error.py
@@ -373,6 +373,6 @@ def calc_quantile_crps_sum(
     CRPS = torch.tensor(0.0)
     for i in range(len(quantiles)):
         q_pred = torch.quantile(predictions.sum(-1), quantiles[i], dim=1)
-        q_loss = calc_quantile_loss(targets, q_pred, quantiles[i], masks)
+        q_loss = calc_quantile_loss(q_pred, targets, quantiles[i], masks)
         CRPS += q_loss / denominator
     return CRPS.item() / len(quantiles)