forked from klmilam/columbia-ad-17
-
Notifications
You must be signed in to change notification settings - Fork 0
/
hptuning.yaml
54 lines (54 loc) · 1.33 KB
/
hptuning.yaml
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
trainingInput:
runtimeVersion: '1.14'
pythonVersion: '3.5'
hyperparameters:
goal: MAXIMIZE
enableTrialEarlyStopping: true
hyperparameterMetricTag: mean_per_class_accuracy
maxTrials: 2
maxParallelTrials: 2
params:
- parameterName: learning-rate
type: DOUBLE
minValue: .00000001 # YAML floats should include a `.`
maxValue: .0001
scaleType: UNIT_LINEAR_SCALE
- parameterName: beta
type: DOUBLE
minValue: .95
maxValue: 1.0
scaleType: UNIT_LINEAR_SCALE
- parameterName: first-layer-size
type: DISCRETE
discreteValues:
- 256
- 512
- 1028
scaleType: UNIT_LINEAR_SCALE
- parameterName: num-layers
type: DISCRETE
discreteValues:
#- 1
- 2
- 3
- 4
scaleType: UNIT_LINEAR_SCALE
- parameterName: layer-sizes-scale-factor
type: DISCRETE
discreteValues:
- .5
- 1
# - 2
scaleType: UNIT_LINEAR_SCALE
- parameterName: reg_rate
type: DOUBLE
minValue: 0.4
maxValue: 0.7
scaleType: UNIT_LINEAR_SCALE
- parameterName: first-filter-size
type: DISCRETE
discreteValues:
- 256
- 512
- 1028
scaleType: UNIT_LINEAR_SCALE