diff --git a/grass7/raster/r.learn.ml2/r.learn.train/r.learn.train.py b/grass7/raster/r.learn.ml2/r.learn.train/r.learn.train.py index 5c7a50c894..9028f0999a 100644 --- a/grass7/raster/r.learn.ml2/r.learn.train/r.learn.train.py +++ b/grass7/raster/r.learn.ml2/r.learn.train/r.learn.train.py @@ -627,7 +627,7 @@ def main(): if any(param_grid) is True: if group_id is None and mode == "classification": - inner = StratifiedKFold(n_splits=2, random_state=random_state) + inner = StratifiedKFold(n_splits=2, random_state=random_state, shuffle=True) elif group_id is None and mode == "regression": inner = KFold(n_splits=2, random_state=random_state) else: @@ -638,7 +638,7 @@ def main(): # outer resampling method (cv=cv) if cv > 1: if group_id is None and mode == "classification": - outer = StratifiedKFold(n_splits=cv, random_state=random_state) + outer = StratifiedKFold(n_splits=cv, random_state=random_state, shuffle=True) elif group_id is None and mode == "regression": outer = KFold(n_splits=cv, random_state=random_state) else: