Non-linear non autoregressive experiments

2024-05-04 16:51:26 +02:00
parent e0c42797e0
commit 75b35bb2c9
14 changed files with 527 additions and 335 deletions
--- a/src/trainers/quantile_trainer.py
+++ b/src/trainers/quantile_trainer.py
@@ -915,9 +915,9 @@ class NonAutoRegressiveQuantileRegression(Trainer):
                targets = targets.squeeze(-1).cpu().numpy()  # (batch_size, 96)

                for i, q in enumerate(quantiles):
-                    quantile_counter[q] += np.sum(targets < outputs[:, i, :])
+                    quantile_counter[q] += np.sum(targets < outputs[:, :, i])

-                total += len(targets)
+                total += len(targets) * 96

        # to numpy array of length len(quantiles)
        percentages = np.array([quantile_counter[q] / total for q in quantiles])