Quarter embedding using trigonometry + more thesis writing

2024-04-17 21:48:13 +02:00
parent 6b02c9aab8
commit 8fb2a7fc7b
18 changed files with 3467 additions and 55 deletions
--- a/src/trainers/trainer.py
+++ b/src/trainers/trainer.py
@@ -262,13 +262,11 @@ class Trainer:
        self.model.eval()

        # set full day skip
-        _, test_loader = self.data_processor.get_dataloaders(
+        train_loader, test_loader = self.data_processor.get_dataloaders(
            predict_sequence_length=self.model.output_size
        )

-        # if not hasattr(self, "plot_quantile_percentages"):
-        #     self.log_final_metrics(task, train_loader, train=True)
-
+        self.log_final_metrics(task, train_loader, train=True)
        self.log_final_metrics(task, test_loader, train=False)

    def test(self, test_loader: torch.utils.data.DataLoader):