config.toml 619 B

1234567891011121314151617181920212223
  1. # Pipline Scenario to run
  2. # "scen_train_all" - Train and evaluate model, with noise
  3. # "scen_load_all" - Load and evaluate model, with noise
  4. # "scen_load_eval" - Load and evaluate model, without noise
  5. scenario = "scen_train_all"
  6. [data]
  7. mri_files_path = "../data/PET_volumes_customtemplate_float32/"
  8. xls_file_path = "../data/LP_ADNIMERGE.csv"
  9. seed = 42
  10. data_splits = [0.7, 0.2, 0.1] # train, validation, test
  11. image_channels = 1
  12. clin_data_channels = 2
  13. num_classes = 2 # AD, NL
  14. [training]
  15. device = "cuda:0" # "cpu", "cuda", "mps"
  16. batch_size = 32
  17. ensemble_size = 50
  18. droprate = 0.05
  19. learning_rate = 0.0001
  20. num_epochs = 30