|
|
@@ -6,18 +6,18 @@ xls_data = '/export/home/nschense/alzheimers/alzheimers_nn/LP_ADNIMERGE.csv'
|
|
|
model_output = '/export/home/nschense/alzheimers/alzheimers_nn/saved_models/'
|
|
|
|
|
|
[training]
|
|
|
-device = 'cuda:1'
|
|
|
-runs = 50
|
|
|
-max_epochs = 30
|
|
|
+device = 'cuda:1'
|
|
|
+runs = 50 # Number of models
|
|
|
+max_epochs = 30 # Epochs per model
|
|
|
|
|
|
[dataset]
|
|
|
-excluded_ids = [91, 108, 268, 269, 272, 279, 293, 296, 307]
|
|
|
+excluded_ids = [91, 108, 268, 269, 272, 279, 293, 296, 307] # Invalid images to exclude from the dataset
|
|
|
validation_split = 0.4 #Splits the dataset into the train and validation/test set, 50% each for validation and test
|
|
|
#|---TEST---|---VALIDATION---|---TRAIN---|
|
|
|
#|splt*0.5 | split*0.5 | 1-split |
|
|
|
|
|
|
[model]
|
|
|
-name = 'cnn-50x30'
|
|
|
+name = 'cnn-50x30' # Ensemble name - will be folder for saved ensemble
|
|
|
image_channels = 1
|
|
|
clin_data_channels = 2
|
|
|
|
|
|
@@ -28,7 +28,7 @@ droprate = 0.5
|
|
|
|
|
|
[operation]
|
|
|
silent = false
|
|
|
-exclude_blank_ids = false
|
|
|
+exclude_blank_ids = false # Determines if invalid IDs are excluded or not
|
|
|
|
|
|
[ensemble]
|
|
|
name = 'cnn-50x30'
|
