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

Add ALSTM config

Browse Source
This commit is contained in:
Jactus
2020-11-25 19:29:30 +08:00
parent 05599d1de8
commit a99db6a1dc
10 changed files with 139 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/workflow_by_code_hats.py
View File

@@ -100,7 +100,7 @@ if __name__ == "__main__":
# model = train_model(task)
model = init_instance_by_config(task["model"])
dataset = init_instance_by_config(task["dataset"])
model.fit(dataset,save_path='benchmarks/HATS/model_hat.pkl')
model.fit(dataset, save_path="benchmarks/HATS/model_hat.pkl")
pred_score = model.predict(dataset)