This commit is contained in:
gorold
2022-08-15 11:37:06 +08:00
parent 9f6ccba9a3
commit 36eb5e3005
+1 -1
View File
@@ -49,7 +49,7 @@ class ForecastDataset(Dataset):
f"only 'lookback_len' xor 'lookback_mult' should be specified"
self.flag = flag
sself.lookback_len = int(horizon_len * lookback_mult) if lookback_mult is not None else lookback_len
self.lookback_len = int(horizon_len * lookback_mult) if lookback_mult is not None else lookback_len
self.lookback_aux_len = lookback_aux_len
self.horizon_len = horizon_len
self.scale = scale