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

Update Dropna function in processor and check model loading and saving with pickle.

Browse Source
This commit is contained in:
lwwang1995
2020-11-13 10:55:10 +08:00
parent 59745486f5
commit 138ab10c1a
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -112,12 +112,12 @@ class Fillna(Processor):
"""Process infinity """ """Process infinity """
def __call__(self, df): def __call__(self, df):
def fill_na(df): def fill_na(df, columns=None, fill=0):
for col in df.columns:
# FIXME: Such behavior is very weird if columns == None:
df[col] = df[col].fillna(0) columns = df.columns
df[columns] = df[columns].fillna(fill)
df.sort_index(inplace=True)
return df return df
return fill_na(df) return fill_na(df)