From 18bf4b547731c946d6c7cfc05955c039e8943706 Mon Sep 17 00:00:00 2001 From: Young Date: Thu, 8 Apr 2021 03:52:58 +0000 Subject: [PATCH] parameter adjustment --- qlib/model/ens/group.py | 4 ++-- qlib/workflow/task/collect.py | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/qlib/model/ens/group.py b/qlib/model/ens/group.py index 9cc5db971..d138b917c 100644 --- a/qlib/model/ens/group.py +++ b/qlib/model/ens/group.py @@ -63,7 +63,7 @@ class RollingGroup(Group): grouped_dict.setdefault(key[:-1], {})[key[-1]] = values return grouped_dict - def __init__(self, group_func=None, ens: Ensemble = RollingEnsemble()): - super().__init__(group_func=group_func, ens=ens) + def __init__(self, group_func=None): + super().__init__(group_func=group_func, ens=RollingEnsemble()) if group_func is None: self.group = RollingGroup.rolling_group diff --git a/qlib/workflow/task/collect.py b/qlib/workflow/task/collect.py index 63d4a6a04..6b9418daf 100644 --- a/qlib/workflow/task/collect.py +++ b/qlib/workflow/task/collect.py @@ -28,19 +28,19 @@ class RecorderCollector(Collector): def __init__( self, exp_name, - artifacts_path={"pred": "pred.pkl", "IC": "sig_analysis/ic.pkl"}, rec_key_func=None, - artifacts_key=None, rec_filter_func=None, + artifacts_path={"pred": "pred.pkl", "IC": "sig_analysis/ic.pkl"}, + artifacts_key=None, ): """init RecorderCollector Args: exp_name (str): the name of Experiment - artifacts_path (dict, optional): The artifacts name and its path in Recorder. Defaults to {"pred": "pred.pkl", "IC": "sig_analysis/ic.pkl"}. rec_key_func (Callable): a function to get the key of a recorder. If None, use recorder id. - artifacts_key (str or List, optional): the artifacts key you want to get. If None, get all artifacts. rec_filter_func (Callable, optional): filter the recorder by return True or False. Defaults to None. + artifacts_path (dict, optional): The artifacts name and its path in Recorder. Defaults to {"pred": "pred.pkl", "IC": "sig_analysis/ic.pkl"}. + artifacts_key (str or List, optional): the artifacts key you want to get. If None, get all artifacts. """ self.exp_name = exp_name self.artifacts_path = artifacts_path