From 138ab10c1ab306bf78f6f4372ba411faea97a4e1 Mon Sep 17 00:00:00 2001 From: lwwang1995 Date: Fri, 13 Nov 2020 10:55:10 +0800 Subject: [PATCH] Update Dropna function in processor and check model loading and saving with pickle. --- qlib/data/dataset/processor.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/qlib/data/dataset/processor.py b/qlib/data/dataset/processor.py index 66db55323..308c531b9 100755 --- a/qlib/data/dataset/processor.py +++ b/qlib/data/dataset/processor.py @@ -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)