|
1 | 1 | # see https://cran.r-project.org/web/packages/ParamHelpers/ParamHelpers.pdfmakeNum
|
2 |
| -# the parameter names should match names of the arguments expected by |
3 |
| -# the benchmark |
| 2 | +# the parameter names should match names of the arguments expected by the benchmark |
| 3 | + |
4 | 4 | param.set <- makeParamSet(
|
5 | 5 | makeDiscreteParam("batch_size", values = c(16, 32, 64, 128, 256, 512)),
|
6 |
| - makeIntegerParam("epochs", lower = 1, upper = 2), |
7 |
| - makeDiscreteParam("activation", values = c("softmax","elu","softplus","softsign", |
8 |
| - "relu", "tanh","sigmoid","hard_sigmoid", |
9 |
| - "linear")), |
10 |
| - makeDiscreteParam("dense", values = c("500 100 50", "1000 500 100 50", |
11 |
| - "2000 1000 500 100 50")), |
12 |
| - makeDiscreteParam("optimizer", values = c("adam", "sgd", "rmsprop","adagrad", |
13 |
| - "adadelta","adamax","nadam")), |
14 |
| - makeNumericParam("drop", lower = 0, upper = 0.5), |
15 |
| - makeDiscreteParam("conv", values = c("50 50 50 50 50 1", |
16 |
| - "25 25 25 25 25 1", |
17 |
| - "64 32 16 32 64 1", |
18 |
| - "100 100 100 100 100 1", |
19 |
| - "32 20 16 32 10 1")) |
| 6 | + makeIntegerParam("epochs", lower = 5, upper = 500), |
| 7 | + makeDiscreteParam("activation", values = c("softmax", "elu", "softplus", "softsign", "relu", "tanh", "sigmoid", "hard_sigmoid", "linear")), |
| 8 | + makeDiscreteParam("dense", values = c("500 100 50", "1000 500 100 50", "2000 1000 500 100 50", "2000 1000 1000 500 100 50", "2000 1000 1000 1000 500 100 50")), |
| 9 | + makeDiscreteParam("optimizer", values = c("adam", "sgd", "rmsprop", "adagrad", "adadelta","adamax","nadam")), |
| 10 | + makeNumericParam("drop", lower = 0, upper = 0.9), |
| 11 | + makeNumericParam("learning_rate", lower = 0.00001, upper = 0.1), |
| 12 | + makeDiscreteParam("conv", values = c("50 50 50 50 50 1", "25 25 25 25 25 1", "64 32 16 32 64 1", "100 100 100 100 100 1", "32 20 16 32 10 1")) |
20 | 13 | ## DEBUG PARAMETERS: DON'T USE THESE IN PRODUCTION RUN
|
21 | 14 | ## makeDiscreteParam("conv", values = c("32 20 16 32 10 1"))
|
22 | 15 | )
|
23 |
| - |
24 |
| - |
25 |
| - |
26 |
| - |
0 commit comments