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

fix format

Browse Source
This commit is contained in:
Young
2020-11-26 12:42:20 +00:00
parent 47f416f1dc
commit 6b053137fd
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_gats.py
View File

@@ -169,7 +169,7 @@ class GAT(Model):
daily_shuffle = list(zip(daily_index, daily_count)) daily_shuffle = list(zip(daily_index, daily_count))
np.random.shuffle(daily_shuffle) np.random.shuffle(daily_shuffle)
daily_index, daily_count = zip(*daily_shuffle) daily_index, daily_count = zip(*daily_shuffle)
return daily_index, daily_count return daily_index, daily_count
def train_epoch(self, x_train, y_train): def train_epoch(self, x_train, y_train):

2
qlib/contrib/model/pytorch_hats.py
View File

@@ -175,7 +175,7 @@ class HATS(Model):
daily_shuffle = list(zip(daily_index, daily_count)) daily_shuffle = list(zip(daily_index, daily_count))
np.random.shuffle(daily_shuffle) np.random.shuffle(daily_shuffle)
daily_index, daily_count = zip(*daily_shuffle) daily_index, daily_count = zip(*daily_shuffle)
return daily_index, daily_count return daily_index, daily_count
def train_epoch(self, x_train, y_train): def train_epoch(self, x_train, y_train):