DIA
/
enose_2025
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

patch

main
ifiguero 2025-03-18 10:26:48 -03:00
parent 8de4ee579c
commit 3557df0648
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
TrainerClass.py
View File

@ -539,7 +539,7 @@ class eNoseTrainer:
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('{}/{}/{}-w{}'.format(self.name, self.loader.target, dataset, window), exist_ok=True)
os.makedirs('{}/{}/{}_{}-w{}'.format(self.name, self.loader.target, dataset, filter, window), exist_ok=True)
X_train, X_test = X_conv1d[train_index], X_conv1d[test_index]
Y_train, Y_test = Y_conv1d[train_index], Y_conv1d[test_index]
G_train, G_test = G_conv1d[train_index], G_conv1d[test_index]
@ -556,12 +556,15 @@ class eNoseTrainer:
self.bar.update()
continue
model_file = '{}/{}/{}_{}-w{}/{}'.format(self.name, self.loader.target, dataset, filter, window, model_id )
X_train_sample, _, Y_train_sample, _ = train_test_split(X_train, Y_train, stratify=G_train, train_size=0.8*sample_size / len(X_train), random_state=get_seed())
X_test_sample, _, Y_test_sample, _ = train_test_split(X_test, Y_test, stratify=G_test, train_size=0.2*sample_size / len(X_test), random_state=get_seed())
self.logger.debug(f"Y_train_sample: {Y_train_sample.shape}")
self.logger.debug(f"Y_test_sample: {Y_test_sample.shape}")
best_model, study, best_batch_size = self.search_best_conv1D_v1(X_train_sample, X_test_sample, Y_train_sample, Y_test_sample, epochs=10, num_trials=10)
# Save study results to an Excel file
trials_data = []
@ -606,7 +609,7 @@ class eNoseTrainer:
"mae": mae_test,
"rmse": rmse_test
}] )
self.ledger = pd.concat([self.ledger, newrow], ignore_index=True)
self.bar.update()
self.saveCheckPoint()