config.toml 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  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. # "scen_analyze" - Analyze existing evaluation files only (no training/eval)
  6. scenario = "scen_train_all"
  7. [run]
  8. # Script/automation options. Leave both false for normal interactive use.
  9. autorun = false # start the pipeline automatically once the config loads
  10. autoquit = false # exit the app when the pipeline finishes (0/1 return code)
  11. # In autorun, allow overwriting PROTECTED outputs (models/evaluations) without a
  12. # prompt. Not needed for "scen_analyze", which only writes regeneratable analysis
  13. # artifacts and therefore never triggers the overwrite guard.
  14. force_overwrite = false
  15. [data]
  16. mri_files_path = "../data/PET_volumes_customtemplate_float32/"
  17. xls_file_path = "../data/LP_ADNIMERGE.csv"
  18. seed = 42
  19. data_splits = [0.7, 0.2, 0.1] # train, validation, test
  20. image_channels = 1
  21. clin_data_channels = 2
  22. num_classes = 2 # AD, NL
  23. [training]
  24. device = "cuda:0" # "cpu", "cuda", "mps"
  25. batch_size = 32
  26. ensemble_size = 30
  27. droprate = 0.05
  28. learning_rate = 0.0001
  29. num_epochs = 25
  30. deterministic = false # force deterministic cuDNN kernels (slower, exact reproduction)
  31. [evaluation]
  32. # Gaussian noise standard deviations applied to images during noisy evaluation
  33. # (step 6). 0.0 is the clean baseline and should be kept first.
  34. noise_levels = [0.0, 0.02, 0.05, 0.1, 0.2]
  35. # Monte-Carlo forward passes used to estimate Bayesian predictive/model
  36. # uncertainty (step 5/6). Ignored for the deterministic ensemble.
  37. mc_passes = 30