data/models/prothgt-config-biological-process.yaml

lr:
  desc: null
  value: 0.001
wd:
  desc: null
  value: 5.0e-05
epochs:
  desc: null
  value: 100
tr_batch_size:
  desc: null
  value: 1024
hidden_channels:
  desc: null
  value:
  - 128
  - - 256
    - 128
    - 64
    - 1
num_heads:
  desc: null
  value: 8
num_layers:
  desc: null
  value: 2
num_neighbors:
  desc: null
  value:
  - 256
mlp_dropout:
  desc: null
  value: 0.4
mlp_lr:
  desc: null
  value: 0.005
mlp_wd:
  desc: null
  value: 0.0001
neg_sample_ratio:
  desc: null
  value: 100
pos_weight:
  desc: null
  value: 100