| 1234567891011121314151617181920 |
- [data]
- mri_files_path = "../data/PET_volumes_customtemplate_float32/"
- xls_file_path = "../data/LP_ADNIMERGE.csv"
- seed = 1
- data_splits = [0.7, 0.2, 0.1] # train, validation, test
- image_channels = 1
- clin_data_channels = 2
- num_classes = 2 # AD, NL
- [training]
- device = "mps" # "cpu", "cuda", "mps"
- batch_size = 32
- ensemble_size = 1
- droprate = 0.1
- learning_rate = 0.001
- num_epochs = 1
- [output]
- path = "../models/"
|
