forked from sukrutrao/Attribution-Evaluation
-
Notifications
You must be signed in to change notification settings - Fork 0
/
configs.py
41 lines (41 loc) · 825 Bytes
/
configs.py
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
"""
Specify different configurations for each attribution method in this file.
For RISE, specify the path to save and load the random masks.
"""
attributor_configs = {
"RISE": {
"test": {"n": 1000, "mask_path": "/path/to/save/and/load/rise/masks/"},
},
"Occlusion": {
"Occ5_2": {"ks": 5,
"stride": 2},
"Occ16_8": {"ks": 16, "stride": 8},
},
"IxG": {
"test": {}
},
"GB": {
"test": {"apply_abs": True}
},
"Grad": {
"test": {"abs": True},
},
"IntGrad": {
"test": {}
},
"GradCam": {
"test": {},
},
"LayerCam": {
"test": {},
},
"GradCamPlusPlus": {
"test": {},
},
"ScoreCam": {
"test": {},
},
"AblationCam": {
"test": {},
},
}