wassname/seq2seq-time
1
0
Fork 0
mirror of https://github.com/wassname/seq2seq-time.git synced 2026-07-06 05:16:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

misc

Browse Source
This commit is contained in:
wassname
2020-10-27 06:43:50 +08:00
parent 6eda47b76f
commit 052fd6596c
9 changed files with 2920 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
seq2seq_time/data/dataset.py
+2 -2
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.dropna(subset=columns_target)
self.df = df.dropna(subset=columns_target).ffill()
self.window_past = window_past
self.window_future = window_future
@@ -100,7 +100,7 @@ class Seq2SeqDataSet(torch.utils.data.Dataset):
def __repr__(self):
t = self.df.index
return f'<{type(self).__name__}(shape={self.df.shape}, times={t[0]} to {t[1]} at {t.freq.freqstr})>'
return f'<{type(self).__name__}(shape={self.df.shape}, times={t[0]} to {t[1]})>'
class Seq2SeqDataSets(torch.utils.data.Dataset):