Skip to content

Commit 69848ef

Browse files
committed
InstanceHardnessThreshold StratifiedKFold bugfix
1 parent c3900a6 commit 69848ef

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

imblearn/under_sampling/instance_hardness_threshold.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,8 @@ def _sample(self, X, y):
231231
"""
232232

233233
# Create the different folds
234-
skf = list(StratifiedKFold(y, n_folds=self.cv, shuffle=False,
234+
235+
skf = list(StratifiedKFold(y, self.cv, shuffle=False,
235236
random_state=self.random_state))
236237

237238
probabilities = np.zeros(y.shape[0], dtype=float)

0 commit comments

Comments
 (0)