Merge pull request #63 from i4Ds/feature/limb-config

adds a new limb config, an analysis notebook and changes the scaling …

.vscode/launch.json

@@ -58,7 +58,21 @@
         "--config-file=./config/ce-vae/run-1.yaml"
       ]
     },
-
+    {
+      "name": "Python: SDO CLI Predict Threshold",
+      "type": "python",
+      "request": "launch",
+      "stopOnEntry": false,
+      "justMyCode": false,
+      "program": "${workspaceRoot}/src/sdo/cli.py",
+      "console": "integratedTerminal",
+      "args": [
+        "sood",
+        "threshold",
+        "predict",
+        "--config-file=./config/ce-vae/run-1.yaml"
+      ]
+    },
     {
       "name": "Python: SDO CLI Generate",
       "type": "python",

config/ce-vae/run-fhnw-full-2-256.yaml

@@ -11,9 +11,9 @@ data:
   num_data_loader_workers:
     value: 16
   prefetch_factor:
-    value: 4
+    value: 6
   batch_size:
-    value: 32
+    value: 8
   sdo_ml_v2:
     train_year:
       value:
@@ -57,7 +57,7 @@ train:
   early_stopping:
     value: true
   use_geco:
-    value: true
+    value: false
 devices:
   gpus:
     value:

config/ce-vae/run-fhnw-full-2-limb.yaml

@@ -0,0 +1,66 @@
+model:
+  load_path:
+    value: null
+data:
+  data_dir:
+    value: /mnt/nas05/astrodata01/astroml_data/sdomlv2_full/sdomlv2.zarr
+  dataset:
+    value: SDOMLDatasetV2
+  num_data_loader_workers:
+    value: 16
+  prefetch_factor:
+    value: 6
+  batch_size:
+    value: 8
+  sdo_ml_v2:
+    train_year:
+      value:
+        - "2010"
+        - "2012"
+        - "2013"
+        - "2015"
+        - "2017"
+        - "2018"
+        - "2020"
+    train_start_date:
+      value: null
+    train_end_date:
+      value: null
+    test_year:
+      value:
+        - "2011"
+        - "2014"
+        - "2016"
+        - "2019"
+    test_start_date:
+      value: null
+    test_end_date:
+      value: null
+    train_val_split_ratio:
+      value: 0.9
+    train_val_split_temporal_chunk_size:
+      value: 14d
+    sampling_strategy:
+      value: "chunk"
+    mask_limb:
+      value: true
+    mask_limb_radius_scale_factor:
+      value: 0.9
+predict:
+  mode:
+    value: sample
+  pred_dir:
+    value: "./output/predictions"
+log_dir:
+  value: "./output/train-sdo-ml"
+train:
+  n_epochs:
+    value: 20
+  early_stopping:
+    value: true
+  use_geco:
+    value: false
+devices:
+  gpus:
+    value:
+      - 2