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

black format

Browse Source
This commit is contained in:
bxdd
2021-03-25 19:59:22 +08:00
parent 68246b3b6d
commit e119c8576c
1 changed files with 26 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
examples/rolling_process_data/workflow.py
View File

@@ -13,6 +13,7 @@ from qlib.contrib.data.handler import Alpha158
from qlib.utils import exists_qlib_data, init_instance_by_config
from qlib.tests.data import GetData
class RollingDataWorkflow(object):
MARKET = "csi300"
@@ -52,12 +53,12 @@ class RollingDataWorkflow(object):
self._dump_pre_handler("pre_handler.py")
pre_handler = self._load_pre_handler("pre_handler.py")
train_start_time = (2010,1,1)
train_end_time = (2012,12,31)
valid_start_time = (2013,1,1)
valid_end_time = (2013,12,31)
test_start_time = (2014,1,1)
test_end_time = (2014,12,31)
train_start_time = (2010, 1, 1)
train_end_time = (2012, 12, 31)
valid_start_time = (2013, 1, 1)
valid_end_time = (2013, 12, 31)
test_start_time = (2014, 1, 1)
test_end_time = (2014, 12, 31)
dataset_config = {
"class": "DatasetH",
@@ -71,9 +72,9 @@ class RollingDataWorkflow(object):
"end_time": datetime(*test_end_time),
"fit_start_time": datetime(*train_start_time),
"fit_end_time": datetime(*train_end_time),
"data_loader_kwargs":{
"data_loader_kwargs": {
"handler_config": pre_handler,
}
},
},
},
"segments": {
@@ -95,9 +96,18 @@ class RollingDataWorkflow(object):
"end_time": datetime(test_end_time[0] + 1, *test_end_time[1:]),
},
segment_kwargs={
"train": (datetime(train_start_time[0] + 1, *train_start_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:])),
"train": (
datetime(train_start_time[0] + 1, *train_start_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:]),
),
},
)