liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 08:16:54 +08:00
Code Activity

Fix bugs for Gats model.

Browse Source
This commit is contained in:
lwwang1995
2020-12-07 23:07:33 +08:00
committed by you-n-g
parent 666e1ffcbd
commit dcfa8110e8
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
qlib/contrib/model/pytorch_gats_ts.py
View File

@@ -37,10 +37,12 @@ class DailyBatchSampler(Sampler):
self.data_source = data_source self.data_source = data_source
self.data = self.data_source.data.loc[self.data_source.get_index()] self.data = self.data_source.data.loc[self.data_source.get_index()]
self.daily_count = self.data.groupby(level=0).size().values[1:] self.daily_count = self.data.groupby(level=0).size().values # calculate number of samples in each batch
self.daily_index = np.roll(np.cumsum(self.daily_count), 1)[1:] self.daily_index = np.roll(np.cumsum(self.daily_count), 1) # calculate begin index of each batch
self.daily_index[0] = 0
def __iter__(self): def __iter__(self):
for idx, count in zip(self.daily_index, self.daily_count): for idx, count in zip(self.daily_index, self.daily_count):
yield np.arange(idx, idx + count) yield np.arange(idx, idx + count)