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

balck format

Browse Source
This commit is contained in:
bxdd
2021-03-25 19:56:22 +08:00
parent efe134e9f4
commit a04c6bd6c9
1 changed files with 26 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
examples/rolling_process_data/workflow.py
View File

@@ -12,6 +12,7 @@ from qlib.contrib.data.handler import Alpha158
from qlib.utils import exists_qlib_data, init_instance_by_config from qlib.utils import exists_qlib_data, init_instance_by_config
from qlib.tests.data import GetData from qlib.tests.data import GetData
class RollingDataWorkflow(object): class RollingDataWorkflow(object):
MARKET = "csi300" MARKET = "csi300"
@@ -72,7 +73,7 @@ class RollingDataWorkflow(object):
"fit_end_time": datetime(*train_end_time), "fit_end_time": datetime(*train_end_time),
"data_loader_kwargs": { "data_loader_kwargs": {
"handler_config": pre_handler, "handler_config": pre_handler,
} },
}, },
}, },
"segments": { "segments": {
@@ -94,9 +95,18 @@ class RollingDataWorkflow(object):
"end_time": datetime(test_end_time[0] + 1, *test_end_time[1:]), "end_time": datetime(test_end_time[0] + 1, *test_end_time[1:]),
}, },
segment_kwargs={ segment_kwargs={
"train": (datetime(train_start_time[0] + 1, *train_start_time[1:]), datetime(train_end_time[0], *train_end_time[1:])), "train": (
"valid": (datetime(valid_start_time[0] + 1, *valid_start_time[1:]), datetime(valid_end_time[0], *valid_end_time[1:])), datetime(train_start_time[0] + 1, *train_start_time[1:]),
"test": (datetime(test_start_time[0] + 1, *test_start_time[1:]), datetime(test_end_time[0], *test_end_time[1:])), datetime(train_end_time[0], *train_end_time[1:]),
),
"valid": (
datetime(valid_start_time[0] + 1, *valid_start_time[1:]),
datetime(valid_end_time[0], *valid_end_time[1:]),
),
"test": (
datetime(test_start_time[0] + 1, *test_start_time[1:]),
datetime(test_end_time[0], *test_end_time[1:]),
),
}, },
) )
@@ -147,4 +157,3 @@ if __name__ == "__main__":
} }
dataset = init_instance_by_config(task["dataset"]) dataset = init_instance_by_config(task["dataset"])