-
Notifications
You must be signed in to change notification settings - Fork 33
/
experiment_config.json
120 lines (120 loc) · 2.84 KB
/
experiment_config.json
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
{
"model_names": [
"lr", "ann"
],
"data_names": [
"adult", "compas", "gaussian", "german",
"gmsc", "heart", "heloc", "pima"
],
"methods": [
"control", "grad", "ig", "itg", "sg", "shap", "lime"
],
"metrics": [
"PRA", "RC",
"FA", "RA", "SA", "SRA",
"PGU", "PGI",
"RIS", "RRS", "ROS"
],
"n_test_samples": 1000,
"explainers": {
"control": {
"seed": 0
},
"grad": {
"absolute_value": false
},
"ig": {
"method": "gausslegendre",
"multiply_by_inputs": false
},
"itg": {},
"sg": {
"n_samples": 100,
"standard_deviation": 0.1,
"seed": 0
},
"shap": {
"n_samples": 500,
"model_impl": "torch",
"seed": 0
},
"lime": {
"n_samples": 1000,
"kernel_width": 0.75,
"std": 0.1,
"mode": "tabular",
"sample_around_instance": true,
"discretize_continuous": false,
"seed": 0
}
},
"evaluators": {
"ground_truth_metrics": {
"k": 0.25,
"AUC": true
},
"prediction_metrics": {
"k": 0.25,
"AUC": true,
"std": 0.1,
"n_samples": 100,
"seed": -1,
"n_jobs": -1
},
"stability_metrics": {
"std": 1e-5,
"n_samples": 1000,
"n_perturbations": 100,
"p_norm": 2,
"seed": -1,
"n_jobs": -1
}
},
"training": {
"epochs": 100,
"learning_rate": 0.001,
"scaler": "minmax",
"seed": 0,
"warmup": 5,
"adult": {
"batch_size": 256,
"pos_class_weight": 0.55,
"mean_pred_bound": 0.15
},
"compas": {
"batch_size": 32,
"pos_class_weight": 0.4,
"mean_pred_bound": 0.93
},
"gaussian": {
"batch_size": 32,
"pos_class_weight": 0.5,
"mean_pred_bound": 0.4
},
"german": {
"batch_size": 16,
"pos_class_weight": 0.4,
"mean_pred_bound": 0.9
},
"gmsc": {
"batch_size": 256,
"pos_class_weight": 0.25,
"mean_pred_bound": 0.96
},
"heart": {
"batch_size": 16,
"pos_class_weight": 0.75,
"mean_pred_bound": 0.1
},
"heloc": {
"batch_size": 32,
"pos_class_weight": 0.5,
"mean_pred_bound": 0.4
},
"pima": {
"batch_size": 16,
"pos_class_weight": 0.65,
"mean_pred_bound": 0.3
}
}
}