shufle
parent
7f1f29d78f
commit
550066ee5f
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue