|
|
@@ -1,6 +1,6 @@
|
|
|
[data]
|
|
|
-mri_files_path = "../data/PET_volumes_customtemplate_float32/"
|
|
|
-xls_file_path = "../data/LP_ADNIMERGE.csv"
|
|
|
+mri_files_path = "/data/data_wnx1/_Data/AlzheimersDL/CNN+RNN-2class-1cnn+data/PET_volumes_customtemplate_float32/"
|
|
|
+xls_file_path = "/export/home/nschense/alzheimers/alzheimers_nn/LP_ADNIMERGE.csv"
|
|
|
seed = 1
|
|
|
data_splits = [0.7, 0.2, 0.1] # train, validation, test
|
|
|
image_channels = 1
|
|
|
@@ -9,12 +9,12 @@ num_classes = 2 # AD, NL
|
|
|
|
|
|
|
|
|
[training]
|
|
|
-device = "mps" # "cpu", "cuda", "mps"
|
|
|
+device = "cuda:1" # "cpu", "cuda", "mps"
|
|
|
batch_size = 32
|
|
|
-ensemble_size = 1
|
|
|
+ensemble_size = 5
|
|
|
droprate = 0.1
|
|
|
learning_rate = 0.001
|
|
|
-num_epochs = 1
|
|
|
+num_epochs = 10
|
|
|
|
|
|
[output]
|
|
|
-path = "../models/"
|
|
|
+path = "../models/5x10/"
|
