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

again

main
ifiguero 2025-03-16 22:38:24 -03:00
parent 99ae1569e3
commit be4ac3f26c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TrainerClass.py
View File

@ -236,10 +236,10 @@ class eNoseTrainer:
'dense_units': tune.choice([32, 64, 128, 256]), 'dense_units': tune.choice([32, 64, 128, 256]),
'dropout': tune.choice([0.05, 0.15, 0.3]), 'dropout': tune.choice([0.05, 0.15, 0.3]),
'lr': tune.choice([0.01, 0.005, 0.001]), 'lr': tune.choice([0.01, 0.005, 0.001]),
'batch_size': tune.choice([16, 32, 64]), 'batch_size': tune.choice([16, 32, 64, 128]),
'epochs': epochs 'epochs': epochs
} }
total_space = (3*3*3*2*3*3*3*3) total_space = (3*3*3*2*4*3*3*4)
self.logger.info(f"total_space: {total_space}, num_samples: {int(nsamples*total_space)}") self.logger.info(f"total_space: {total_space}, num_samples: {int(nsamples*total_space)}")