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:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user