iter

glemaitre · glemaitre · commit ec4a717c2c83 · 2020-11-08T15:49:47.000+01:00
diff --git a/imblearn/base.py b/imblearn/base.py
@@ -273,8 +273,9 @@ def fit_resample(self, X, y):
                 X, y, accept_sparse=self.accept_sparse
             )
 
+        self._classes_counts = get_classes_counts(y)
         self.sampling_strategy_ = check_sampling_strategy(
-            self.sampling_strategy, y, self._sampling_type
+            self.sampling_strategy, self._classes_counts, self._sampling_type
         )
 
         output = self._fit_resample(X, y)

Original file line number	Diff line number	Diff line change
`@@ -273,8 +273,9 @@ def fit_resample(self, X, y):`
`273`	`273`	`X, y, accept_sparse=self.accept_sparse`
`274`	`274`	`)`
`275`	`275`
	`276`	`+ self._classes_counts = get_classes_counts(y)`
`276`	`277`	`self.sampling_strategy_ = check_sampling_strategy(`
`277`		`- self.sampling_strategy, y, self._sampling_type`
	`278`	`+ self.sampling_strategy, self._classes_counts, self._sampling_type`
`278`	`279`	`)`
`279`	`280`
`280`	`281`	`output = self._fit_resample(X, y)`