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:35:29 -05:00
parent ac8968c3e3
commit 0ef149f76b
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

@ -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):