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

fix

main
ifiguero 2025-03-17 15:40:01 -03:00
parent df2c1f3775
commit 52949f6aab
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TrainerClass.py
View File

@ -347,7 +347,7 @@ class eNoseTrainer:
plt.close()
def fit(self):
windows = [128,]
windows = [64, 128, 256,]
# windows = [128, 256, 384]
total_train_queue = 2*int(1/self.ratio)*(len(self.get_model_train())+len(windows))
self.logger.info("{:=^60}".format(f'Begin Fit {total_train_queue} Models'))
@ -518,7 +518,7 @@ class eNoseTrainer:
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)
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=32)
# Save study results to an Excel file
trials_data = []
for trial in study.trials:
@ -605,7 +605,7 @@ class eNoseTrainer:
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)
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=32)
# Save study results to an Excel file
trials_data = []
for trial in study.trials: