forked from wj210/Causal-Faithfulness
-
Notifications
You must be signed in to change notification settings - Fork 0
/
get_score.sh
40 lines (31 loc) · 890 Bytes
/
get_score.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
model_name='gemma2-9B'
dataset_name='comve'
## Get scores ##
for model_name in gemma2-2B gemma2-2B-chat gemma2-9B gemma2-9B-chat gemma2-27B gemma2-27B-chat
do
python score_utils/compute_scores.py \
--dataset_name csqa esnli comve \
--metric causal cf_edit ccshap \
--model_name $model_name \
--causal_type STR STR_10
done
## Plot ##
for model_name in gemma2-2B gemma2-2B-chat gemma2-9B gemma2-9B-chat gemma2-27B gemma2-27B-chat
do
python score_utils/plot.py \
--dataset_name $dataset_name \
--model_name $model_name
done
## OOD only for ComVE ##
## OOD GN ##
python score_utils/compute_scores.py \
--dataset_name $dataset_name \
--metric ood_GN \
--model_name 'gemma2-2B-chat' \
--causal_type STR STR_10
## OOD SHAP ##
python score_utils/compute_scores.py \
--dataset_name $dataset_name \
--metric ood \
--model_name 'gemma2-2B-chat' \
--causal_type STR STR_10