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

Merge branch 'main' of github.com:you-n-g/qlib into main

Browse Source
This commit is contained in:
Dong Zhou
2020-11-24 10:09:33 +08:00
parent 7f9f54faf4 0112d06ed5
commit 641e7f38bc
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"""