shufle
parent
7f1f29d78f
commit
550066ee5f
|
@ -204,7 +204,7 @@ class eNoseTrainer:
|
||||||
node = os.uname()[1]
|
node = os.uname()[1]
|
||||||
X_xboost, Y_xboost, G_xboost = self.loader.load_dataset_xboost()
|
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'
|
dataset = 'Tabular'
|
||||||
os.makedirs('{}/{}/{}'.format(self.name, self.target, dataset), exist_ok=True)
|
os.makedirs('{}/{}/{}'.format(self.name, self.target, dataset), exist_ok=True)
|
||||||
|
|
Loading…
Reference in New Issue