liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-06 20:41:09 +08:00
Code Activity

rename group -> fields_group

Browse Source
This commit is contained in:
Young
2020-11-24 02:05:05 +00:00
parent f536241a5f
commit 0112d06ed5
3 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
qlib/data/dataset/processor.py
View File

@@ -74,16 +74,16 @@ class Processor(Serializable):
class DropnaProcessor(Processor):
def __init__(self, group=None):
self.group = group
def __init__(self, fields_group=None):
self.fields_group = fields_group
def __call__(self, df):
return df.dropna(subset=get_group_columns(df, self.group))
return df.dropna(subset=get_group_columns(df, self.fields_group))
class DropnaLabel(DropnaProcessor):
def __init__(self, group="label"):
super().__init__(group=group)
def __init__(self, fields_group="label"):
super().__init__(fields_group=fields_group)
def is_for_infer(self) -> bool:
"""The samples are dropped according to label. So it is not usable for inference"""