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 14:09:38 -05:00
parent 452014ec17
commit 1178cb7343
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pipeline.py
View File

@ -239,7 +239,7 @@ class Components :
real_df = pd.DataFrame() real_df = pd.DataFrame()
if x_cols : if x_cols :
args['data'] = args['data'][list(set(args['data'].columns) - set(x_cols))] args['data'] = args['data'][list(set(args['data'].columns) - set(x_cols))]
real_df = args['data'][x_cols].copy() real_df = df['data'][x_cols].copy()
args['candidates'] = 1 if 'candidates' not in args else int(args['candidates']) args['candidates'] = 1 if 'candidates' not in args else int(args['candidates'])
if 'gpu' in args : if 'gpu' in args :