|
2 | 2 | # the parameter names should match names of the arguments expected by
|
3 | 3 | # the benchmark
|
4 | 4 | param.set <- makeParamSet(
|
5 |
| - makeDiscreteParam("batch_size", values = c(32, 64, 128)), |
6 |
| - makeIntegerParam("epochs", lower = 1, upper = 3) |
| 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("optimizer", values = c("adam", "sgd", "rmsprop","adagrad", |
| 11 | + "adadelta")), |
| 12 | + makeNumericParam("learning_rate", lower = 0.001, upper = 0.2), |
| 13 | + makeNumericParam("noise", lower = 0.001, upper = 0.5) |
| 14 | + #makeDiscreteParam("num_hidden", values = c("512 256 128 64 32 16", "512 64")), |
| 15 | + #makeDiscreteParam("molecular_num_hidden", values = c("54 12", "32 16 8")) |
| 16 | + ## DEBUG PARAMETERS: DON'T USE THESE IN PRODUCTION RUN |
| 17 | + ## makeDiscreteParam("conv", values = c("32 20 16 32 10 1")) |
7 | 18 | )
|
| 19 | + |
| 20 | + |
| 21 | + |
| 22 | + |
0 commit comments