bug fix: fields skipped (training)

4 years ago · 0ef149f76b
parent ac8968c3e3
commit 0ef149f76b
1 changed files with 1 additions and 1 deletions
--- a/pipeline.py
+++ b/pipeline.py
@ -79,7 +79,7 @@ class Components :
 		if 'columns' in _args and 'data' in _args :
 			_df = _args['data']
 			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 []

 	def train(self,**args):