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