diff --git a/TrainerClass.py b/TrainerClass.py index f6027e8..570089b 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -204,7 +204,7 @@ class eNoseTrainer: node = os.uname()[1] X_xboost, Y_xboost, G_xboost = self.loader.load_dataset_xboost() - gss = StratifiedGroupKFold(n_splits=int(1/self.ratio), random_state=get_seed()) + gss = StratifiedGroupKFold(n_splits=int(1/self.ratio), shuffle=True, random_state=get_seed()) dataset = 'Tabular' os.makedirs('{}/{}/{}'.format(self.name, self.target, dataset), exist_ok=True)