-
Notifications
You must be signed in to change notification settings - Fork 1
/
definitions.py
57 lines (54 loc) · 1.01 KB
/
definitions.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
classification = {
"Alpha" : 0,
"Beta" : 1,
"Chi" : 2,
"Delta" : 3,
"Epsilon" : 4,
"Eta" : 5,
"Gamma" : 6,
"Iota" : 7,
"Kappa" : 8,
"Lambda" : 9,
"Mu" : 10,
"Nu" : 11,
"Omega" : 12,
"Omicron" : 13,
"Phi" : 14,
"Pi" : 15,
"Psi" : 16,
"Rho" : 17,
"Sigma" : 18,
"Tau" : 19,
"Theta" : 20,
"Upsilon" : 21,
"Xi" :22,
"Zeta" : 23
}
PARAMETERS = {
"HiddenLayers": set([2, 4, 8, 16, 32, 50]),
"LearningRate": set([0.001, 0.01, 0.03, 0.05]),
"Optimizer": set(["Adam", "RMSprop"]),
"NumFilters": set([16, 32, 64, 96]),
"Activation": set(["relu"]),
"KernelSize": set([3]),
"Momentum": set([.9]),
"Epochs": set([75]),
"BatchSize": set([512])
}
OPTMZ_ARGS = {
"Adam": {
"learning_rate": "LearningRate"
},
"RMSprop": {
"learning_rate": "LearningRate",
"momentum": "Momentum"
}
}
ROWS = 70
COLS = 70
CHANNELS = 3
CLASSES = 24
LENGTH = 1000
BLOCKS = 50
EPOCHS = 75
N_JOBS = 1