iter

Author: glemaitre

imblearn/over_sampling/_smote/cluster.py

@@ -220,6 +220,8 @@ def _fit_resample(self, X, y):
 
                 # the cluster is already considered balanced
                 if cluster_class_mean < balance_threshold:
+                    print(cluster_class_mean)
+                    print(balance_threshold)
                     continue
 
                 # not enough samples to apply SMOTE

imblearn/over_sampling/_smote/tests/test_kmeans_smote.py

@@ -91,8 +91,6 @@ def test_sample_kmeans_custom(data, k_neighbors, kmeans_estimator):
 def test_sample_kmeans_not_enough_clusters(data):
     X, y = data
     smote = KMeansSMOTE(
-        density_exponent=2,
-        cluster_balance_threshold=1,
         random_state=42,
         kmeans_estimator=10,
         k_neighbors=2,