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

Remove batchsize and add daily-batch mode

Browse Source
This commit is contained in:
meng-ustc
2020-11-26 20:19:23 +08:00
parent a108f753d5
commit 99efaadd38
3 changed files with 32 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/benchmarks/GATs/workflow_config_gats.yaml
View File

@@ -36,7 +36,6 @@ task:
n_epochs: 200
lr: 1e-3
early_stop: 20
batch_size: 800
metric: loss
loss: mse
base_model: LSTM

4
examples/workflow_by_code_gats.py
View File

@@ -66,13 +66,12 @@ if __name__ == "__main__":
"n_epochs": 200,
"lr": 1e-3,
"early_stop": 20,
"batch_size": 800,
"metric": "loss",
"loss": "mse",
"base_model": "LSTM",
"with_pretrain": True,
"seed": 0,
"GPU": 0,
"GPU": "0",
},
},
"dataset": {
@@ -95,7 +94,6 @@ if __name__ == "__main__":
# "record": ['SignalRecord', 'SigAnaRecord', 'PortAnaRecord'],
}
# model = train_model(task)
model = init_instance_by_config(task["model"])
dataset = init_instance_by_config(task["dataset"])
model.fit(dataset)