Steve L. Nyemba 4 years ago
parent 1178cb7343
commit 6bbe767714

@ -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 = df['data'][x_cols].copy() real_df = df[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 :
@ -279,7 +279,7 @@ class Components :
if x_cols : if x_cols :
for _col in x_cols : for _col in x_cols :
if real_df[_col].unique().size > 0 : if real_df[_col].unique().size > 0 :
_df[_col] = self.approximate(df[_col].fillna(-1)) _df[_col] = self.approximate(real_df[_col].fillna(-1))
else: else:
_df[_col] = -1 _df[_col] = -1

Loading…
Cancel
Save