bug fix: fields skipped (training)

dev
Steve L. Nyemba 4 years ago
parent ac8968c3e3
commit 0ef149f76b

@ -79,7 +79,7 @@ class Components :
if 'columns' in _args and 'data' in _args : if 'columns' in _args and 'data' in _args :
_df = _args['data'] _df = _args['data']
terms = _args['columns'] terms = _args['columns']
return [name for name in _df.columns if name in terms] return [name for name in _df.columns if np.sum( [int(field in name )for field in terms ]) > 0 ]
return [] return []
def train(self,**args): def train(self,**args):

Loading…
Cancel
Save