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

...

This commit is contained in:
Steve Nyemba 2021-04-01 12:43:09 -05:00
parent cf478016b0
commit 732ccb42e5
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pipeline.py
View File

@ -136,6 +136,8 @@ class Components :
# We need to make sure that continuous columns are removed # We need to make sure that continuous columns are removed
if x_cols : if x_cols :
_args['data'] = df[list(set(df.columns) - set(x_cols))] _args['data'] = df[list(set(df.columns) - set(x_cols))]
if 'gpu' in args :
_args['gpu'] = args['gpu']
data.maker.train(**_args) data.maker.train(**_args)
if 'autopilot' in ( list(args.keys())) : if 'autopilot' in ( list(args.keys())) :