1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-06 05:51:17 +08:00

update workflow

This commit is contained in:
bxdd
2021-03-25 19:56:04 +08:00
parent 4ec300787e
commit efe134e9f4
3 changed files with 5 additions and 9 deletions

View File

@@ -12,17 +12,15 @@ class RollingDataHandler(DataHandlerLP):
learn_processors=[],
fit_start_time=None,
fit_end_time=None,
data_loader_kwargs={}
data_loader_kwargs={},
):
infer_processors = check_transform_proc(infer_processors, fit_start_time, fit_end_time)
learn_processors = check_transform_proc(learn_processors, fit_start_time, fit_end_time)
data_loader = {
"class": "DataLoaderDH",
"kwargs": {
**data_loader_kwargs
},
}
"kwargs": {**data_loader_kwargs},
}
super().__init__(
instruments=None,

View File

@@ -101,7 +101,7 @@ class RollingDataWorkflow(object):
)
dtrain, dvalid, dtest = dataset.prepare(["train", "valid", "test"])
if __name__ == "__main__":

View File

@@ -258,9 +258,7 @@ class DataLoaderDH(DataLoader):
self.handlers = init_instance_by_config(handler_config, accept_types=DataHandler)
self.is_group = is_group
self.fetch_kwargs = {
"col_set":DataHandler.CS_RAW
}
self.fetch_kwargs = {"col_set": DataHandler.CS_RAW}
self.fetch_kwargs = {**self.fetch_kwargs, **fetch_kwargs}
def load(self, instruments=None, start_time=None, end_time=None) -> pd.DataFrame: