🐛 fix error in golden features, add better subsampling (#201)

mljar · Sep 21, 2020 · ce85f8d · ce85f8d
1 parent 1edf319
commit ce85f8d
Showing 1 changed file with 0 additions and 12 deletions.
diff --git a/supervised/preprocessing/goldenfeatures_transformer.py b/supervised/preprocessing/goldenfeatures_transformer.py
@@ -250,16 +250,4 @@ def _subsample(self, X, y):
                 X, y, train_size=train_size, shuffle=shuffle, stratify=stratify
             )
 
-        """
-        middle, stop = 2500, 5000
-        if X.shape[0] < 5000:
-            stop = X.shape[0]
-            middle = int(stop / 2)
-
-        X_train = X[:middle]
-        X_test = X[middle:stop]
-        y_train = y[:middle]
-        y_test = y[middle:stop]
-        """
-
         return X_train, X_test, y_train, y_test