data/models/prothgt-config-molecular-function.yaml

lr:
  desc: null
  value: 0.005
wd:
  desc: null
  value: 0
epochs:
  desc: null
  value: 100
tr_batch_size:
  desc: null
  value: 512
hidden_channels:
  desc: null
  value:
  - 32
  - - 64
    - 32
    - 16
    - 8
    - 1
num_heads:
  desc: null
  value: 2
num_layers:
  desc: null
  value: 2
num_neighbors:
  desc: null
  value:
  - 32
mlp_dropout:
  desc: null
  value: 0.4
mlp_lr:
  desc: null
  value: 0.01
mlp_wd:
  desc: null
  value: 1.0e-05
neg_sample_ratio:
  desc: null
  value: 100
pos_weight:
  desc: null
  value: 100