Spaces:

lablab-ai-amd-developer-hackathon
/

BrainConnect-ASD

Running

App Files Files Community

Yatsuiii commited on 1 day ago

Commit

8e0b796

verified ·

1 Parent(s): 753a844

Upload app.py with huggingface_hub

Browse files

Files changed (1) hide show

app.py +60 -44

app.py CHANGED Viewed

@@ -28,26 +28,26 @@ _ATLAS_CFG = {
         "net_bounds":  [0, 38, 69, 99, 137, 165, 180, 200],
         "net_colors":  ["#e63946", "#f4a261", "#457b9d", "#2dc653", "#a8dadc", "#8b5cf6", "#6b7280"],
         "ckpts": {
-            "CALTECH":  Path("checkpoints/cc200_caltech.ckpt"),
-            "CMU":      Path("checkpoints/cc200_cmu.ckpt"),
-            "KKI":      Path("checkpoints/cc200_kki.ckpt"),
-            "LEUVEN_1": Path("checkpoints/cc200_leuven_1.ckpt"),
-            "LEUVEN_2": Path("checkpoints/cc200_leuven_2.ckpt"),
-            "MAX_MUN":  Path("checkpoints/cc200_max_mun.ckpt"),
-            "NYU":      Path("checkpoints/cc200_nyu.ckpt"),
-            "OHSU":     Path("checkpoints/cc200_ohsu.ckpt"),
-            "OLIN":     Path("checkpoints/cc200_olin.ckpt"),
-            "PITT":     Path("checkpoints/cc200_pitt.ckpt"),
-            "SBL":      Path("checkpoints/cc200_sbl.ckpt"),
-            "SDSU":     Path("checkpoints/cc200_sdsu.ckpt"),
-            "STANFORD": Path("checkpoints/cc200_stanford.ckpt"),
-            "TRINITY":  Path("checkpoints/cc200_trinity.ckpt"),
-            "UCLA_1":   Path("checkpoints/cc200_ucla_1.ckpt"),
-            "UCLA_2":   Path("checkpoints/cc200_ucla_2.ckpt"),
-            "UM_1":     Path("checkpoints/cc200_um_1.ckpt"),
-            "UM_2":     Path("checkpoints/cc200_um_2.ckpt"),
-            "USM":      Path("checkpoints/cc200_usm.ckpt"),
-            "YALE":     Path("checkpoints/cc200_yale.ckpt"),
         },
     },
     "aal": {
@@ -60,26 +60,26 @@ _ATLAS_CFG = {
         "net_bounds":  [0, 20, 34, 50, 68, 80, 92, 116],
         "net_colors":  ["#457b9d", "#2dc653", "#8b5cf6", "#e63946", "#f4a261", "#6b7280", "#a8dadc"],
         "ckpts": {
-            "CALTECH":  Path("checkpoints/aal_caltech.ckpt"),
-            "CMU":      Path("checkpoints/aal_cmu.ckpt"),
-            "KKI":      Path("checkpoints/aal_kki.ckpt"),
-            "LEUVEN_1": Path("checkpoints/aal_leuven_1.ckpt"),
-            "LEUVEN_2": Path("checkpoints/aal_leuven_2.ckpt"),
-            "MAX_MUN":  Path("checkpoints/aal_max_mun.ckpt"),
-            "NYU":      Path("checkpoints/aal_nyu.ckpt"),
-            "OHSU":     Path("checkpoints/aal_ohsu.ckpt"),
-            "OLIN":     Path("checkpoints/aal_olin.ckpt"),
-            "PITT":     Path("checkpoints/aal_pitt.ckpt"),
-            "SBL":      Path("checkpoints/aal_sbl.ckpt"),
-            "SDSU":     Path("checkpoints/aal_sdsu.ckpt"),
-            "STANFORD": Path("checkpoints/aal_stanford.ckpt"),
-            "TRINITY":  Path("checkpoints/aal_trinity.ckpt"),
-            "UCLA_1":   Path("checkpoints/aal_ucla_1.ckpt"),
-            "UCLA_2":   Path("checkpoints/aal_ucla_2.ckpt"),
-            "UM_1":     Path("checkpoints/aal_um_1.ckpt"),
-            "UM_2":     Path("checkpoints/aal_um_2.ckpt"),
-            "USM":      Path("checkpoints/aal_usm.ckpt"),
-            "YALE":     Path("checkpoints/aal_yale.ckpt"),
         },
     },
     "ho": {
@@ -89,10 +89,26 @@ _ATLAS_CFG = {
         "net_bounds":  [0, 18, 30, 48, 68, 80, 96, 111],
         "net_colors":  ["#457b9d", "#2dc653", "#e63946", "#f4a261", "#6b7280", "#a8dadc", "#8b5cf6"],
         "ckpts": {
-            "NYU":  Path("checkpoints/ho_nyu.ckpt"),
-            "USM":  Path("checkpoints/ho_usm.ckpt"),
-            "UCLA": Path("checkpoints/ho_ucla.ckpt"),
-            "UM":   Path("checkpoints/ho_um.ckpt"),
         },
     },
 }

         "net_bounds":  [0, 38, 69, 99, 137, 165, 180, 200],
         "net_colors":  ["#e63946", "#f4a261", "#457b9d", "#2dc653", "#a8dadc", "#8b5cf6", "#6b7280"],
         "ckpts": {
+            "CALTECH":  Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_caltech/brain-gcn-epoch=020-val_auc=0.953.ckpt"),
+            "CMU":      Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_cmu/brain-gcn-epoch=001-val_auc=0.893.ckpt"),
+            "KKI":      Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_kki/brain-gcn-epoch=014-val_auc=0.917.ckpt"),
+            "LEUVEN_1": Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_leuven_1/brain-gcn-epoch=004-val_auc=0.917.ckpt"),
+            "LEUVEN_2": Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_leuven_2/brain-gcn-epoch=005-val_auc=0.888.ckpt"),
+            "MAX_MUN":  Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_max_mun/brain-gcn-epoch=005-val_auc=0.858.ckpt"),
+            "NYU":      Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_nyu/brain-gcn-epoch=067-val_auc=0.964.ckpt"),
+            "OHSU":     Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_ohsu/brain-gcn-epoch=004-val_auc=0.858.ckpt"),
+            "OLIN":     Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_olin/brain-gcn-epoch=003-val_auc=0.970.ckpt"),
+            "PITT":     Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_pitt/brain-gcn-epoch=009-val_auc=0.935.ckpt"),
+            "SBL":      Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_sbl/brain-gcn-epoch=021-val_auc=0.876.ckpt"),
+            "SDSU":     Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_sdsu/brain-gcn-epoch=001-val_auc=0.864.ckpt"),
+            "STANFORD": Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_stanford/brain-gcn-epoch=002-val_auc=0.923.ckpt"),
+            "TRINITY":  Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_trinity/brain-gcn-epoch=006-val_auc=0.888.ckpt"),
+            "UCLA_1":   Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_ucla_1/brain-gcn-epoch=054-val_auc=0.976.ckpt"),
+            "UCLA_2":   Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_ucla_2/brain-gcn-epoch=055-val_auc=0.863.ckpt"),
+            "UM_1":     Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_um_1/brain-gcn-epoch=013-val_auc=0.959.ckpt"),
+            "UM_2":     Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_um_2/brain-gcn-epoch=005-val_auc=0.899.ckpt"),
+            "USM":      Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_usm/brain-gcn-epoch=020-val_auc=0.970.ckpt"),
+            "YALE":     Path("checkpoints/cc200/adv_brain_mode_k32_site_cc200_loso_yale/brain-gcn-epoch=055-val_auc=0.964.ckpt"),
         },
     },
     "aal": {
         "net_bounds":  [0, 20, 34, 50, 68, 80, 92, 116],
         "net_colors":  ["#457b9d", "#2dc653", "#8b5cf6", "#e63946", "#f4a261", "#6b7280", "#a8dadc"],
         "ckpts": {
+            "CALTECH":  Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_caltech/brain-gcn-epoch=003-val_auc=0.822.ckpt"),
+            "CMU":      Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_cmu/brain-gcn-epoch=004-val_auc=0.775.ckpt"),
+            "KKI":      Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_kki/brain-gcn-epoch=022-val_auc=0.834.ckpt"),
+            "LEUVEN_1": Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_leuven_1/brain-gcn-epoch=001-val_auc=0.858.ckpt"),
+            "LEUVEN_2": Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_leuven_2/brain-gcn-epoch=007-val_auc=0.846.ckpt"),
+            "MAX_MUN":  Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_max_mun/brain-gcn-epoch=056-val_auc=0.769.ckpt"),
+            "NYU":      Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_nyu/brain-gcn-epoch=011-val_auc=0.740.ckpt"),
+            "OHSU":     Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_ohsu/brain-gcn-epoch=006-val_auc=0.799.ckpt"),
+            "OLIN":     Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_olin/brain-gcn-epoch=008-val_auc=0.846.ckpt"),
+            "PITT":     Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_pitt/brain-gcn-epoch=001-val_auc=0.888.ckpt"),
+            "SBL":      Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_sbl/brain-gcn-epoch=018-val_auc=0.828.ckpt"),
+            "SDSU":     Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_sdsu/brain-gcn-epoch=005-val_auc=0.746.ckpt"),
+            "STANFORD": Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_stanford/brain-gcn-epoch=002-val_auc=0.852.ckpt"),
+            "TRINITY":  Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_trinity/brain-gcn-epoch=001-val_auc=0.834.ckpt"),
+            "UCLA_1":   Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_ucla_1/brain-gcn-epoch=000-val_auc=0.846.ckpt"),
+            "UCLA_2":   Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_ucla_2/brain-gcn-epoch=000-val_auc=0.813.ckpt"),
+            "UM_1":     Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_um_1/brain-gcn-epoch=051-val_auc=0.828.ckpt"),
+            "UM_2":     Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_um_2/brain-gcn-epoch=001-val_auc=0.822.ckpt"),
+            "USM":      Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_usm/brain-gcn-epoch=006-val_auc=0.805.ckpt"),
+            "YALE":     Path("checkpoints/aal/adv_brain_mode_k32_site_aal_loso_yale/brain-gcn-epoch=054-val_auc=0.870.ckpt"),
         },
     },
     "ho": {
         "net_bounds":  [0, 18, 30, 48, 68, 80, 96, 111],
         "net_colors":  ["#457b9d", "#2dc653", "#e63946", "#f4a261", "#6b7280", "#a8dadc", "#8b5cf6"],
         "ckpts": {
+            "CALTECH":  Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_caltech/brain-gcn-epoch=013-val_auc=0.888.ckpt"),
+            "CMU":      Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_cmu/brain-gcn-epoch=011-val_auc=0.852.ckpt"),
+            "KKI":      Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_kki/brain-gcn-epoch=059-val_auc=0.917.ckpt"),
+            "LEUVEN_1": Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_leuven_1/brain-gcn-epoch=021-val_auc=0.899.ckpt"),
+            "LEUVEN_2": Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_leuven_2/brain-gcn-epoch=055-val_auc=0.905.ckpt"),
+            "MAX_MUN":  Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_max_mun/brain-gcn-epoch=003-val_auc=0.882.ckpt"),
+            "NYU":      Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_nyu/brain-gcn-epoch=017-val_auc=0.882.ckpt"),
+            "OHSU":     Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_ohsu/brain-gcn-epoch=010-val_auc=0.882.ckpt"),
+            "OLIN":     Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_olin/brain-gcn-epoch=024-val_auc=0.929.ckpt"),
+            "PITT":     Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_pitt/brain-gcn-epoch=018-val_auc=0.882.ckpt"),
+            "SBL":      Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_sbl/brain-gcn-epoch=003-val_auc=0.893.ckpt"),
+            "SDSU":     Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_sdsu/brain-gcn-epoch=095-val_auc=0.935.ckpt"),
+            "STANFORD": Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_stanford/brain-gcn-epoch=002-val_auc=0.888.ckpt"),
+            "TRINITY":  Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_trinity/brain-gcn-epoch=021-val_auc=0.864.ckpt"),
+            "UCLA_1":   Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_ucla_1/brain-gcn-epoch=009-val_auc=0.817.ckpt"),
+            "UCLA_2":   Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_ucla_2/brain-gcn-epoch=001-val_auc=0.797.ckpt"),
+            "UM_1":     Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_um_1/brain-gcn-epoch=005-val_auc=0.852.ckpt"),
+            "UM_2":     Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_um_2/brain-gcn-epoch=006-val_auc=0.870.ckpt"),
+            "USM":      Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_usm/brain-gcn-epoch=000-val_auc=0.840.ckpt"),
+            "YALE":     Path("checkpoints/ho/adv_brain_mode_k32_site_ho_loso_yale/brain-gcn-epoch=004-val_auc=0.876.ckpt"),
         },
     },
 }