Skip to content
Prev Previous commit
Next Next commit
removed non-binary exception
  • Loading branch information
Andrea Lorenzon committed Sep 15, 2020
commit 9ac27974e3f9088769be2ca0207daacad3da70fc
4 changes: 2 additions & 2 deletions imblearn/over_sampling/_rose.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,8 +107,8 @@ def _make_samples(self,

def _fit_resample(self, X, y):

if type_of_target(y) != "binary":
raise Exception("ROSE supports only binary outcome datasets.")
# if type_of_target(y) != "binary":
# raise Exception("ROSE supports only binary outcome datasets.")

X_resampled = np.empty((0, X.shape[1]), dtype=X.dtype)
y_resampled = np.empty((0), dtype=X.dtype)
Expand Down