fix
parent
ac1c6aecfa
commit
6af8ab89e4
|
@ -450,7 +450,7 @@ class eNoseTrainer:
|
|||
Y_discrete = np.sum(Y_discrete, axis=1)
|
||||
|
||||
dataset = f'Conv1D-w{window}'
|
||||
for i, (train_index, test_index) in enumerate(gss.split(X_Conv1d, Y_discrete, G_conv1d)):
|
||||
for i, (train_index, test_index) in enumerate(gss.split(X_conv1d, Y_discrete, G_conv1d)):
|
||||
self.logger.info("{:=^60}".format(f'CV {i+1}/{int(1/self.ratio)} {dataset}'))
|
||||
|
||||
os.makedirs('{}/{}/{}'.format(self.name, self.loader.target, dataset), exist_ok=True)
|
||||
|
|
Loading…
Reference in New Issue