diff --git a/pipeline.py b/pipeline.py index a09fbde..9f57d59 100644 --- a/pipeline.py +++ b/pipeline.py @@ -274,16 +274,16 @@ class Components : # _df = _df[list(set(_df.columns) - set(skip_columns))] - - if set(df.columns) & set(_df.columns) : - _columns = set(df.columns) - set(_df.columns) - df = df[_columns] if x_cols : for _col in x_cols : if df[_col].unique().size > 0 : _df[_col] = self.approximate(df[_col].fillna(-1)) else: _df[_col] = -1 + + if set(df.columns) & set(_df.columns) : + _columns = set(df.columns) - set(_df.columns) + df = df[_columns] # # Let us merge the dataset here and and have a comprehensive dataset