Commit ·
817e21e
1
Parent(s): 6cb1a61
updated demo
Browse files- root_gnn_dgl/run_demo.sh +24 -24
root_gnn_dgl/run_demo.sh
CHANGED
|
@@ -5,45 +5,45 @@ datasets=("ttH" "tHjb" "ggF" "VBF" "WH" "ZH" "ttyy" "tttt" "SingleT_schan" "ttba
|
|
| 5 |
chunks=3
|
| 6 |
|
| 7 |
for data in "${datasets[@]}"; do
|
| 8 |
-
python scripts/prep_data.py --config configs/
|
| 9 |
for ((i=0; i<chunks; i++)); do
|
| 10 |
-
python scripts/prep_data.py --config configs/
|
| 11 |
done
|
| 12 |
done
|
| 13 |
|
| 14 |
-
python scripts/training_script.py --config configs/
|
| 15 |
|
| 16 |
# Finetuning
|
| 17 |
|
| 18 |
-
datasets=("
|
| 19 |
chunks=3
|
| 20 |
|
| 21 |
for data in "${datasets[@]}"; do
|
| 22 |
-
python scripts/prep_data.py --config configs/
|
| 23 |
for ((i=0; i<chunks; i++)); do
|
| 24 |
-
python scripts/prep_data.py --config configs/
|
| 25 |
done
|
| 26 |
done
|
| 27 |
|
| 28 |
-
python scripts/training_script.py --config configs/
|
| 29 |
|
| 30 |
|
| 31 |
# Inference
|
| 32 |
|
| 33 |
-
python scripts/inference.py \
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
python scripts/inference.py \
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
|
|
|
| 5 |
chunks=3
|
| 6 |
|
| 7 |
for data in "${datasets[@]}"; do
|
| 8 |
+
python scripts/prep_data.py --config configs/stats_100K/pretraining_multiclass.yaml --dataset "$data" --shuffle_mode --chunk 0
|
| 9 |
for ((i=0; i<chunks; i++)); do
|
| 10 |
+
python scripts/prep_data.py --config configs/stats_100K/pretraining_multiclass.yaml --dataset "$data" --shuffle_mode --chunk "$i"
|
| 11 |
done
|
| 12 |
done
|
| 13 |
|
| 14 |
+
python scripts/training_script.py --config configs/stats_100K/pretraining_multiclass.yaml --preshuffle --nocompile --lazy
|
| 15 |
|
| 16 |
# Finetuning
|
| 17 |
|
| 18 |
+
datasets=("ttH_CP_even" "ttH_CP_odd")
|
| 19 |
chunks=3
|
| 20 |
|
| 21 |
for data in "${datasets[@]}"; do
|
| 22 |
+
python scripts/prep_data.py --config configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml --dataset "$data" --shuffle_mode --chunk 0
|
| 23 |
for ((i=0; i<chunks; i++)); do
|
| 24 |
+
python scripts/prep_data.py --config configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml --dataset "$data" --shuffle_mode --chunk "$i"
|
| 25 |
done
|
| 26 |
done
|
| 27 |
|
| 28 |
+
python scripts/training_script.py --config configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml --preshuffle --nocompile --lazy
|
| 29 |
|
| 30 |
|
| 31 |
# Inference
|
| 32 |
|
| 33 |
+
# python scripts/inference.py \
|
| 34 |
+
# --target "/global/cfs/projectdirs/atlas/joshua/root_gnn/root_gnn_dgl/data/ntuples/Hyy_pretraining/multilabel_10K/ttH_NLO.root" \
|
| 35 |
+
# --destination "/global/cfs/projectdirs/atlas/joshua/GNN4Colliders/root_gnn_dgl/scores/ttH_NLO.root" \
|
| 36 |
+
# --config "configs/demo/finetuning_ttH_CP_Even_vs_Odd.yaml" \
|
| 37 |
+
# --chunks 1 \
|
| 38 |
+
# --chunkno 0 \
|
| 39 |
+
# --write \
|
| 40 |
+
# --branch 'GNN_Score'
|
| 41 |
+
|
| 42 |
+
# python scripts/inference.py \
|
| 43 |
+
# --target "/global/cfs/projectdirs/atlas/joshua/root_gnn/root_gnn_dgl/data/ntuples/Hyy_pretraining/multilabel_10K/ttH_CPodd.root" \
|
| 44 |
+
# --destination "/global/cfs/projectdirs/atlas/joshua/GNN4Colliders/root_gnn_dgl/scores/ttH_CPodd.root" \
|
| 45 |
+
# --config "configs/demo/finetuning_ttH_CP_Even_vs_Odd.yaml" \
|
| 46 |
+
# --chunks 1 \
|
| 47 |
+
# --chunkno 0 \
|
| 48 |
+
# --write \
|
| 49 |
+
# --branch 'GNN_Score'
|