-
Notifications
You must be signed in to change notification settings - Fork 2
/
main.sh
41 lines (34 loc) · 1.16 KB
/
main.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#! /bin/sh
# cd src || exit
#explainer=subgraphX
#dataset_name=bbbp
#sparsity=0.85
#model_name=gat
# explainer=gnn_explainer
# bash run_one_model.sh $explainer $dataset_name $sparsity $model_name
# explainer=gnn_lrp
# bash run_one_model.sh $explainer $dataset_name 0.8 $model_name
# explainer=gnn_lrp
# bash run_one_model.sh $explainer $dataset_name 0.8 $model_name
# explainer=grad_cam
# bash run_one_model.sh $explainer $dataset_name $sparsity $model_name
for explainer in gnn_explainer pgexplainer gnn_lrp grad_cam subgraphx gstarx orphicx graphsvx same
do
for dataset_name in ba_2motifs bbbp graph_sst2 graph_sst5 ba_shapes mutag twitter bace
do
for model_name in gcn gin gat
do
if [ "$explainer" = "subgraphx" ] || [ "$explainer" = "same" ]; then
for sparsity in 0.5 0.6 0.7 0.8
do
bash run_one_model.sh $explainer $dataset_name $sparsity $model_name
done
else
for sparsity in 0.8 0.7 0.6 0.5
do
bash run_one_model.sh $explainer $dataset_name $sparsity $model_name
done
fi
done
done
done