diff --git a/pipeline.py b/pipeline.py index f4db40c..6cb1be9 100644 --- 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):