Fix Remove_100

pycaret · Nov 27, 2020 · b2833e6 · b2833e6
1 parent bf5b9c9
commit b2833e6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pycaret/internal/preprocess.py b/pycaret/internal/preprocess.py
@@ -2779,6 +2779,7 @@ class Remove_100(BaseEstimator, TransformerMixin):
 
     def __init__(self, target):
         self.target = target
+        self.columns_to_drop = []
 
     def fit(self, data, y=None):
         self.fit_transform(data, y=y)