From 0ef149f76b4d17d0b4bd78847b8c9403d895800a Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Tue, 30 Mar 2021 05:35:29 -0500 Subject: [PATCH] bug fix: fields skipped (training) --- pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):