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

misc

Browse Source
This commit is contained in:
wassname
2020-10-18 15:18:11 +08:00
parent 17fb62e766
commit 4453b25000
4 changed files with 1683 additions and 1010 deletions
Download Patch File Download Diff File Expand all files Collapse all files
seq2seq_time/data/dataset.py
+6 -12
View File
@@ -119,18 +119,12 @@ class Seq2SeqDataSets(torch.utils.data.Dataset):
raise IndexError
def get_rows(self, i):
"""
Output pandas dataframes for display purposes.
"""
x_cols = list(self.df.drop(columns=self.columns_target).columns) + ['tsp_days', 'is_past']
x_past, y_past, x_future, y_future = self.get_components(i)
t_past = self.df.index[i:i+self.window_past]
t_future = self.df.index[i+self.window_past:i+self.window_past + self.window_future]
x_past = pd.DataFrame(x_past, columns=x_cols, index=t_past)
x_future = pd.DataFrame(x_future, columns=x_cols, index=t_future)
y_past = pd.DataFrame(y_past, columns=self.columns_target, index=t_past)
y_future = pd.DataFrame(y_future, columns=self.columns_target, index=t_future)
return x_past, y_past, x_future, y_future
l = 0
for d in self.datasets:
l += len(d)
if i < l:
return d.get_rows(i)
raise IndexError
def __len__(self):
l = 0