liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-18 01:44:34 +08:00
Code Activity

revise settings

Browse Source
This commit is contained in:
Hong Zhang
2020-11-27 00:19:23 +08:00
parent 293c405593
commit 5796363ecf
5 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -52,11 +52,11 @@ class HATS(Model):
num_layers=2,
dropout=0.5,
n_epochs=200,
lr=0.01,
metric="IC",
lr=0.0001,
metric="loss",
early_stop=20,
loss="mse",
base_model="GRU",
base_model="LSTM",
with_pretrain=True,
optimizer="adam",
GPU="0",
@@ -180,7 +180,7 @@ class HATS(Model):
def train_epoch(self, x_train, y_train):
x_train_values = x_train.values
y_train_values = np.squeeze(y_train.values) * 100
y_train_values = np.squeeze(y_train.values)
self.HATS_model.train()