diff --git a/pipeline.py b/pipeline.py index e643278..f4db40c 100644 --- a/pipeline.py +++ b/pipeline.py @@ -104,9 +104,6 @@ class Components : # # - if 'ignore' in args and 'columns' in args['ignore'] : - _cols = self.get_ignore(data=df,columns=args['ignore']['columns']) - df = df[ list(set(df.columns)- set(_cols))] # df = df.fillna('') if schema : _schema = [] @@ -125,6 +122,9 @@ class Components : _args = copy.deepcopy(args) # _args['store'] = args['store']['source'] _args['data'] = df + if 'ignore' in args and 'columns' in args['ignore'] : + _cols = self.get_ignore(data=df,columns=args['ignore']['columns']) + _args['data'] = df[ list(set(df.columns)- set(_cols))] data.maker.train(**_args)