bug1

imgremlin · imgremlin · commit ca4556a8c107 · 2025-04-29T22:59:45.000-04:00
diff --git a/imblearn/under_sampling/_prototype_selection/_random_under_sampler.py b/imblearn/under_sampling/_prototype_selection/_random_under_sampler.py
@@ -112,7 +112,7 @@ def _fit_resample(self, X, y):
             if target_class in self.sampling_strategy_.keys():
                 n_samples = self.sampling_strategy_[target_class]
                 index_target_class = random_state.choice(
-                    range(np.count_nonzero(y == target_class)),
+                    range(np.count_nonzero(y != target_class)),
                     size=n_samples,
                     replace=self.replacement,
                 )
@@ -122,7 +122,7 @@ def _fit_resample(self, X, y):
             idx_under = np.concatenate(
                 (
                     idx_under,
-                    np.flatnonzero(y == target_class)[index_target_class],
+                    np.flatnonzero(y != target_class)[index_target_class],
                 ),
                 axis=0,
             )

Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ def _fit_resample(self, X, y):`
`112`	`112`	`if target_class in self.sampling_strategy_.keys():`
`113`	`113`	`n_samples = self.sampling_strategy_[target_class]`
`114`	`114`	`index_target_class = random_state.choice(`
`115`		`- range(np.count_nonzero(y == target_class)),`
	`115`	`+ range(np.count_nonzero(y != target_class)),`
`116`	`116`	`size=n_samples,`
`117`	`117`	`replace=self.replacement,`
`118`	`118`	`)`
`@@ -122,7 +122,7 @@ def _fit_resample(self, X, y):`
`122`	`122`	`idx_under = np.concatenate(`
`123`	`123`	`(`
`124`	`124`	`idx_under,`
`125`		`- np.flatnonzero(y == target_class)[index_target_class],`
	`125`	`+ np.flatnonzero(y != target_class)[index_target_class],`
`126`	`126`	`),`
`127`	`127`	`axis=0,`
`128`	`128`	`)`