exploration/rnn_sweep.yaml
program: exploration/rnn_sweep_train.py
name: rnn-2
description: lowercase lemmas, max_features 8000, min_df .00005
method: bayes
metric:
name: val_accuracy
goal: maximize
parameters:
epochs:
min: 3
max: 10
learning_rate:
min: 0.00005
max: 0.01
batch_size:
values: [256, 128, 64, 32]
weight_decay:
values: [.0005, .005, .05]
dropout:
values: [.2, .3, .4, .5]
early_terminate:
type: hyperband
min_iter: 3
project: allay-ds-23
entity: alexmjn