add comments

jurjen93 · Oct 29, 2024 · 15e3734 · 15e3734
1 parent b03ed3c
commit 15e3734
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/neural_networks/plots/confusion_matrix.py b/neural_networks/plots/confusion_matrix.py
@@ -176,7 +176,9 @@ def get_dataloader(data_root, mode="val", batch_size=32):
     architecture_name = "surf/TransferLearning"
     # Set Device here
     DEVICE = "cuda"
+    # Thresholds to consider for classification
     thresholds = [0.2, 0.3, 0.4, 0.5]
+    # Change to directory of files. Should have subfolders 'continue_val' and 'stop_val'
     data_root = "/scratch-shared/CORTEX/public.spider.surfsara.nl/lofarvwf/jdejong/CORTEX/calibrator_selection_robertjan/cnn_data"
     # Uses cached confusion matrix for testing the plotting functionalities
     if model_name == "surf/dinov2_09739_rotations" and TESTING: