aou/data-maker
aou
/
data-maker
1
1
Fork 0
Code Issues 4 Pull Requests Packages Projects 1 Releases Wiki Activity

bug fix: fields skipped (training)

This commit is contained in:
Steve Nyemba 2021-03-30 05:23:55 -05:00
parent 13053febb7
commit ac8968c3e3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pipeline.py
View File

@ -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('') # df = df.fillna('')
if schema : if schema :
_schema = [] _schema = []
@ -125,6 +122,9 @@ class Components :
_args = copy.deepcopy(args) _args = copy.deepcopy(args)
# _args['store'] = args['store']['source'] # _args['store'] = args['store']['source']
_args['data'] = df _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) data.maker.train(**_args)