training_classifiers/binding_affinity/wt_smiles_unpooled/best_params.json

{
  "lr": 6.714904102732621e-05,
  "weight_decay": 6.94348785472601e-08,
  "dropout": 0.20599610484012826,
  "hidden_dim": 768,
  "n_heads": 4,
  "n_layers": 3,
  "cls_weight": 0.26109289573917854,
  "batch_size": 16
}