1234567891011121314151617181920 |
- [data]
- mri_files_path = "/data/data_wnx1/_Data/AlzheimersDL/CNN+RNN-2class-1cnn+data/PET_volumes_customtemplate_float32/"
- xls_file_path = "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 = "cuda:1" # "cpu", "cuda", "mps"
- batch_size = 32
- ensemble_size = 5
- droprate = 0.1
- learning_rate = 0.001
- num_epochs = 10
- [output]
- path = "../models/5x10/"
|