config.toml 433 B

1234567891011121314151617181920
  1. [data]
  2. mri_files_path = "../data/PET_volumes_customtemplate_float32/"
  3. xls_file_path = "LP_ADNIMERGE.csv"
  4. seed = 42
  5. data_splits = [0.7, 0.2, 0.1] # train, validation, test
  6. image_channels = 1
  7. clin_data_channels = 2
  8. num_classes = 2 # AD, NL
  9. [training]
  10. device = "cuda:0" # "cpu", "cuda", "mps"
  11. batch_size = 32
  12. ensemble_size = 50
  13. droprate = 0.05
  14. learning_rate = 0.0001
  15. num_epochs = 30
  16. [output]
  17. path = "../models/Full_Ensemble(50x30)/"