bug fix: fields skipped (training)

dev
Steve L. Nyemba 4 years ago
parent 13053febb7
commit ac8968c3e3

@ -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)

Loading…
Cancel
Save