liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-13 15:56:57 +08:00
Code Activity

use list_kwargs instead filter_string

Browse Source
This commit is contained in:
chaosyu
2021-07-16 19:44:06 +08:00
committed by you-n-g
parent 28cb827a23
commit 8b715268bd
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
qlib/workflow/task/collect.py
View File

@@ -139,7 +139,7 @@ class RecorderCollector(Collector):
rec_filter_func=None, rec_filter_func=None,
artifacts_path={"pred": "pred.pkl"}, artifacts_path={"pred": "pred.pkl"},
artifacts_key=None, artifacts_key=None,
filter_string: str = "", list_kwargs={},
): ):
""" """
Init RecorderCollector. Init RecorderCollector.
@@ -165,7 +165,7 @@ class RecorderCollector(Collector):
self.rec_key_func = rec_key_func self.rec_key_func = rec_key_func
self.artifacts_key = artifacts_key self.artifacts_key = artifacts_key
self.rec_filter_func = rec_filter_func self.rec_filter_func = rec_filter_func
self.filter_string = filter_string self.list_kwargs = list_kwargs
def collect(self, artifacts_key=None, rec_filter_func=None, only_exist=True) -> dict: def collect(self, artifacts_key=None, rec_filter_func=None, only_exist=True) -> dict:
""" """
@@ -190,7 +190,7 @@ class RecorderCollector(Collector):
collect_dict = {} collect_dict = {}
# filter records # filter records
recs = self.experiment.list_recorders(filter_string=self.filter_string) recs = self.experiment.list_recorders(**self.list_kwargs)
recs_flt = {} recs_flt = {}
for rid, rec in recs.items(): for rid, rec in recs.items():
if rec_filter_func is None or rec_filter_func(rec): if rec_filter_func is None or rec_filter_func(rec):