This commit is contained in:
bxdd 2021-03-25 19:56:04 +08:00
Родитель 4ec300787e
Коммит efe134e9f4
3 изменённых файлов: 5 добавлений и 9 удалений

Просмотреть файл

@ -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,

Просмотреть файл

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

Просмотреть файл

@ -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: