1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-06 05:51:17 +08:00

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

This commit is contained in:
lwwang1995
2020-11-13 10:55:10 +08:00
parent 59745486f5
commit 138ab10c1a

View File

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