exploration/sweep_cnn.yaml
program: exploration/sweep_cnn_train.py
method: bayes
metric:
name: val_accuracy
goal: maximize
parameters:
epochs:
min: 3
max: 15
learning_rate:
min: 0.0001
max: 0.1
optimizer:
values: ["adam", "adadelta", "nadam"]
num_filters:
min: 2
max: 100
base_kernel_size:
min: 2
max: 4
dropout_rate:
min: 0.1
max: 0.3
loss_function:
values: ["binary_crossentropy"]