-
Notifications
You must be signed in to change notification settings - Fork 2
/
observable_names.py
88 lines (79 loc) · 3.28 KB
/
observable_names.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
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
from collections import OrderedDict
DEFAULT_GRP_KEYS = ['early', 'middle', 'deep']
'''The set of observable statistics that have been finite across all models and tasks tested so far.'''
# preserve observable class ordering
ADMISSIBLE_OBS = OrderedDict()
ADMISSIBLE_OBS['weight'] = ['weightabs_mean',
'weightabs_median',
'weightabs_norm',
'weightabs_thirdquartile',
'weightabs_var',
'weightraw_mean',
'weightraw_median',
'weightraw_norm',
'weightraw_thirdquartile',
'weightraw_var',
'weightsq_mean',
'weightsq_median',
'weightsq_thirdquartile']
ADMISSIBLE_OBS['act'] = ['actabs_mean',
'actabs_median',
'actabs_norm',
'actabs_thirdquartile',
'actraw_mean',
'actraw_median',
'actraw_norm',
'actraw_thirdquartile']
ADMISSIBLE_OBS['grad'] = ['gradavgabs_id',
'gradavgraw_id',
'gradavgsq_id']
ADMISSIBLE_OBS['all'] = ADMISSIBLE_OBS['weight'] + ADMISSIBLE_OBS['act'] + ADMISSIBLE_OBS['grad']
'''The set of all observable statistics (not guaranteed to always be finite).'''
# preserve observable class ordering
ALL_OBS = OrderedDict()
ALL_OBS['weight'] = ['weightabs_kurtosis',
'weightabs_mean',
'weightabs_median',
'weightabs_norm',
'weightabs_skew',
'weightabs_thirdquartile',
'weightabs_var',
'weightraw_kurtosis',
'weightraw_mean',
'weightraw_median',
'weightraw_norm',
'weightraw_skew',
'weightraw_thirdquartile',
'weightraw_var',
'weightsq_kurtosis',
'weightsq_mean',
'weightsq_median',
'weightsq_norm',
'weightsq_skew',
'weightsq_thirdquartile',
'weightsq_var']
ALL_OBS['act'] = ['actabs_kurtosis',
'actabs_mean',
'actabs_median',
'actabs_norm',
'actabs_skew',
'actabs_thirdquartile',
'actabs_var',
'actraw_kurtosis',
'actraw_mean',
'actraw_median',
'actraw_norm',
'actraw_skew',
'actraw_thirdquartile',
'actraw_var',
'actsq_kurtosis',
'actsq_mean',
'actsq_median',
'actsq_norm',
'actsq_skew',
'actsq_thirdquartile',
'actsq_var']
ALL_OBS['grad'] = ['gradavgabs_id',
'gradavgraw_id',
'gradavgsq_id']
ALL_OBS['all'] = ALL_OBS['weight'] + ALL_OBS['act'] + ALL_OBS['grad']