fixing demo

root_gnn_dgl/configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml

@@ -29,15 +29,15 @@ Datasets:
   ttH_CP_even: &dataset_defn
     module: root_gnn_base.dataset
     class: LazyDataset
-    shuffle_chunks: 1
+    shuffle_chunks: 3
     batch_size: 1024
     padding_mode: NONE #one of STEPS, FIXED, or NONE
     args: &dataset_args
       name: ttH_CP_even
       label: 0
       # weight_var: weight
-      chunks: 1
-      buffer_size: 0
+      chunks: 3
+      buffer_size: 2
       file_names: ttH_NLO.root
       tree_name: output
       fold_var: Number

root_gnn_dgl/configs/stats_100K/pretraining_multiclass.yaml

@@ -27,14 +27,14 @@ Datasets:
   ttH: &dataset_defn
     module: root_gnn_base.dataset
     class: LazyDataset
-    shuffle_chunks: 1
+    shuffle_chunks: 3
     padding_mode: NONE #one of STEPS, FIXED, or NONE
     args: &dataset_args
       name: ttH
       label: 0
       # weight_var: weight
-      chunks: 1
-      buffer_size: 0
+      chunks: 3
+      buffer_size: 2
       file_names: ttH_NLO_inc.root
       tree_name: output
       fold_var: Number

root_gnn_dgl/configs/stats_100K/ttH_CP_even_vs_odd.yaml

@@ -19,15 +19,15 @@ Datasets:
   ttH_CP_even: &dataset_defn
     module: root_gnn_base.dataset
     class: LazyDataset
-    shuffle_chunks: 1
+    shuffle_chunks: 3
     batch_size: 1024
     padding_mode: NONE #one of STEPS, FIXED, or NONE
     args: &dataset_args
       name: ttH_CP_even
       label: 0
       # weight_var: weight
-      chunks: 1
-      buffer_size: 0
+      chunks: 3
+      buffer_size: 2
       file_names: ttH_NLO.root
       tree_name: output
       fold_var: Number

root_gnn_dgl/run_demo.sh

@@ -19,31 +19,41 @@ datasets=("ttH_CP_even" "ttH_CP_odd")
 chunks=3
 
 for data in "${datasets[@]}"; do
-    python scripts/prep_data.py --config configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml --dataset "$data" --shuffle_mode --chunk 0
+    python scripts/prep_data.py --config configs/stats_100K/ttH_CP_even_vs_odd.yaml --dataset "$data" --shuffle_mode --chunk 0
     for ((i=0; i<chunks; i++)); do
-        python scripts/prep_data.py --config configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml --dataset "$data" --shuffle_mode --chunk "$i"
+        python scripts/prep_data.py --config configs/stats_100K/ttH_CP_even_vs_odd.yaml --dataset "$data" --shuffle_mode --chunk "$i"
     done
 done
 
-python scripts/training_script.py --config configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml --preshuffle --nocompile --lazy
+python scripts/training_script.py --config configs/stats_100K/ttH_CP_even_vs_odd.yaml --preshuffle --nocompile --lazy
 
+python scripts/training_script.py --config configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml --preshuffle --nocompile --lazy
 
 # Inference
-
-# python scripts/inference.py \
-#     --target "/global/cfs/projectdirs/atlas/joshua/root_gnn/root_gnn_dgl/data/ntuples/Hyy_pretraining/multilabel_10K/ttH_NLO.root" \
-#     --destination "/global/cfs/projectdirs/atlas/joshua/GNN4Colliders/root_gnn_dgl/scores/ttH_NLO.root" \
-#     --config "configs/demo/finetuning_ttH_CP_Even_vs_Odd.yaml" \
-#     --chunks 1 \
-#     --chunkno 0 \
-#     --write \
-#     --branch 'GNN_Score'
-
-# python scripts/inference.py \
-#     --target "/global/cfs/projectdirs/atlas/joshua/root_gnn/root_gnn_dgl/data/ntuples/Hyy_pretraining/multilabel_10K/ttH_CPodd.root" \
-#     --destination "/global/cfs/projectdirs/atlas/joshua/GNN4Colliders/root_gnn_dgl/scores/ttH_CPodd.root" \
-#     --config "configs/demo/finetuning_ttH_CP_Even_vs_Odd.yaml" \
-#     --chunks 1 \
-#     --chunkno 0 \
-#     --write \
-#     --branch 'GNN_Score'
+files=(
+    "ttH_NLO.root"
+    "ttH_CPodd.root"
+)
+
+config=(
+    "configs/stats_100K/ttH_CP_even_vs_odd.yaml"
+    "configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml"
+)
+
+branch_name=( 
+    "cp_score"
+    "finetuning_cp_score"
+)
+
+for ((j=0; j<${#files[@]}; j++))
+do
+    python scripts/inference.py \
+        --target "/global/cfs/projectdirs/trn007/lbl_atlas/data/stats_100K/${files[j]}" \
+        --destination "/global/cfs/projectdirs/trn007/lbl_atlas/data/scores/stats_100K/${files[j]}" \
+        --config "${config[@]}" \
+        --branch_name "${branch_name[@]}" \
+        --chunks 1 \
+        --chunkno 0 \
+        --write
+    echo ''
+done