This commit is contained in:
wassname
2020-10-26 15:33:32 +08:00
parent b40d311e0b
commit 6eda47b76f
7 changed files with 8 additions and 15866 deletions
+1 -1
View File
@@ -31,7 +31,7 @@ class Seq2SeqDataSet(torch.utils.data.Dataset):
assert df.index.freq is not None, 'should have freq'
assert_no_objects(df)
self.df = df
self.df = df.dropna(subset=columns_target)
self.window_past = window_past
self.window_future = window_future