This commit is contained in:
wassname
2020-10-19 13:41:12 +08:00
parent 6647ef5e65
commit e986751e41
10 changed files with 5613 additions and 2 deletions
+2 -2
View File
@@ -62,8 +62,8 @@ class Seq2SeqDataSet(torch.utils.data.Dataset):
x_future = x[self.window_past:]
y_future = y[self.window_past:]
# Stop it cheating by using future weather measurements
x_future[:, self._icol_blank] = 0
# Stop it cheating by using future weather measurements. Fill in with last value
x_future[:, self._icol_blank] = x_past[0, self._icol_blank]
return x_past, y_past, x_future, y_future