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

black format

Browse Source
This commit is contained in:
Dong Zhou
2020-11-27 08:58:44 +08:00
parent ca0a689362
commit 6b4156ab9f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -180,7 +180,7 @@ class HATS(Model):
def train_epoch(self, x_train, y_train): def train_epoch(self, x_train, y_train):
x_train_values = x_train.values x_train_values = x_train.values
y_train_values = np.squeeze(y_train.values) y_train_values = np.squeeze(y_train.values)
self.HATS_model.train() self.HATS_model.train()