save

HieuPhan33 · Mar 29, 2024 · 353140b · 353140b
1 parent 9de75f7
commit 353140b
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -26,3 +26,4 @@ Pretrain/.vscode/launch_model.json
 Pretrain/cl_global_1e-1_clean
 # Pretrain
 */__pycached__
+Zero-shot_classification/*csv
diff --git a/Zero-shot_classification/configs/rsna_mavl.yaml b/Zero-shot_classification/configs/rsna_mavl.yaml
@@ -5,7 +5,9 @@ concept_book: '../Pretrain/concept_gen/gpt4_mimic.json'
 dataset: 'rsna'
 root: '/data/data/rsna'
 model: mavl
-model_path: '../checkpoints/checkpoint_full_40.pth'
+# model_path: '../checkpoints/checkpoint_full_40.pth'
+model_path: '../../medslip/checkpoints/cl_global_1e-1_true/checkpoint_40.pth'
+
 
 image_res: 224
 test_batch_size: 512

diff --git a/Zero-shot_classification/result_mavl_siim_feature.csv b/Zero-shot_classification/result_mavl_siim_feature.csv
@@ -0,0 +1,3 @@
+Class Name,Accuracy,Max F1,AUC ROC,Precision,Recall
+pneumothorax,0.8713854121709107,0.7795100222717148,0.9204018547140648,0.75,0.8114374034003091
+Average,0.8713854121709107,0.7795100222717148,0.9204018547140648,0.75,0.8114374034003091
diff --git a/Zero-shot_grounding/configs/rsna_mavl.yaml b/Zero-shot_grounding/configs/rsna_mavl.yaml
@@ -7,7 +7,9 @@ dataset: rsna
 root: /data/data/rsna/
 target: pneumonia
 model: mavl
-model_path: '../checkpoints/checkpoint_full_40.pth'
+# model_path: '../checkpoints/checkpoint_full_40.pth'
+model_path: '../../medslip/checkpoints/cl_global_1e-1_true/checkpoint_40.pth'
+
 
 image_res: 224
 test_batch_size: 512