bishop.yaml

name: adult
method: random
metric:
  goal: maximize
  name: AUC
parameters:
  out_len:
    values:
    - 2
    - 4
    - 8
    - 16
    - 24
    - 32
    - 48
    - 64
    - 128
    - 256
    - 320
  patch_dim:
    values:
    - 1
    - 2
    - 4
    - 6
    - 8
    - 12
    - 16
    - 24
  emb_dim:
    values:
    - 16
    - 24
    - 32
    - 48
    - 64
    - 128
    - 256
    - 320
  n_agg:
    values:
    - 2
    - 3
    - 4
    - 5
    - 6
    - 7
    - 8
  factor:
    values:
    - 5
    - 10
    - 15
  d_model:
    values:
    - 64
    - 128
    - 256
    - 512
    - 1024
  d_ff:
    values:
    - 64
    - 128
    - 256
    - 512
    - 1024
  n_heads:
    values:
    - 2
    - 4
    - 6
    - 8
    - 10
    - 12
  e_layers:
    values:
    - 2
    - 3
    - 4
    - 5
  d_layers:
    values:
    - 0
    - 1
  dropout:
    values:
    - 0
    - 0.1
    - 0.2
    - 0.3
    - 0.4
    - 0.5
  learning_rate:
    distribution: log_uniform_values
    min: 1.0e-06
    max: 0.0001
  seed:
    value: 66
command:
- ${env}
- python
- ${program}
- --data
- adult
- --record
- --sweep
- --train_epochs
- '200'
program: bishop.py