lars/aiRtrafficNN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Debugging of mode 3

Browse Source
This commit is contained in:
lars
2025-11-27 08:52:43 +01:00
parent 2cd518321f
commit 4be4bd79a2
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Code/python/notebooks/hyperparameter_scan.py
+4 -4
View File
@@ -239,14 +239,14 @@ def run_mode_3():
for warm in [300, 600, 900, 1800]: for warm in [300, 600, 900, 1800]:
for pred in [150, 300, 600, 900, 1800]: for pred in [150, 300, 600, 900, 1800]:
start = 30 * 60 - warm start_offset = 30 * 60 - warm
end = 30 * 60 - pred end_offset = 30 * 60 - pred
base_ds = dataloader.SaveDataset( base_ds = dataloader.SaveDataset(
torch.load("long_dataset.pt"), torch.load("long_dataset.pt"),
step=step, step=step,
start_offset=start, start_offset=start_offset,
end_offset=end, end_offset=end_offset,
) )
train_ds, val_ds = split_dataset(base_ds) train_ds, val_ds = split_dataset(base_ds)
Code/python/src/aiRNN/dataloader.py
+3
View File
@@ -143,8 +143,11 @@ class SaveDataset(Dataset):
t0 = self.start_offset t0 = self.start_offset
t1 = max_len - self.end_offset t1 = max_len - self.end_offset
self.idx_feat = torch.arange(t0, self.X_feat.shape[1], self.step) self.idx_feat = torch.arange(t0, self.X_feat.shape[1], self.step)
print(self.idx_feat)
self.idx_time = torch.arange(t0, t1, self.step) self.idx_time = torch.arange(t0, t1, self.step)
print(self.idx_time)
self.idx_out = torch.arange(0, self.Y_out.shape[1] - self.end_offset, self.step) self.idx_out = torch.arange(0, self.Y_out.shape[1] - self.end_offset, self.step)
print(self.idx_out)
def __len__(self): def __len__(self):
return self.X_feat.shape[0] return self.X_feat.shape[0]