From b9147df69ce28576c22d3db12a7fb3fb7cd0991b Mon Sep 17 00:00:00 2001 From: tangxi1227 <412467218@qq.com> Date: Mon, 10 Aug 2020 23:06:49 +0800 Subject: [PATCH] modify cluster_centroids, fix a bug --- .../under_sampling/_prototype_generation/_cluster_centroids.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imblearn/under_sampling/_prototype_generation/_cluster_centroids.py b/imblearn/under_sampling/_prototype_generation/_cluster_centroids.py index ebdbed27b..deef67894 100644 --- a/imblearn/under_sampling/_prototype_generation/_cluster_centroids.py +++ b/imblearn/under_sampling/_prototype_generation/_cluster_centroids.py @@ -171,7 +171,7 @@ def _fit_resample(self, X, y): self.estimator_.set_params(**{"n_clusters": n_samples}) self.estimator_.fit(X[y == target_class]) X_new, y_new = self._generate_sample( - X, y, self.estimator_.cluster_centers_, target_class + X[y == target_class], y[y == target_class], self.estimator_.cluster_centers_, target_class ) X_resampled.append(X_new) y_resampled.append(y_new)