ifiguero 2025-03-18 21:37:05 -03:00
parent ac1c6aecfa
commit 6af8ab89e4
1 changed files with 1 additions and 1 deletions

View File

@ -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)