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

{
  "lr": 0.001,
  "wd": 5e-5,
  "epochs": 100,
  "tr_batch_size": 1024,
  "hidden_channels": [128, [256, 128, 64, 1]],
  "num_heads": 8,                                                                 
  "num_layers": 2,
  "num_neighbors": [256],
  
  "mlp_dropout" : 0.4,
  "mlp_lr": 0.005,
  "mlp_wd": 0.0001,

  "neg_sample_ratio": 100,
  "pos_weight": 100,
  }