{
  "cv_scheme": "Butina T=0.4 ECFP4 GroupKFold, 5 folds",
  "n_seeds": 5,
  "pretrain_epochs": 30,
  "finetune_epochs": 50,
  "head1_weight": 4.0,
  "architecture": {
    "type": "D-MPNN (BondMessagePassing)",
    "hidden_dim": 300,
    "n_layers_ffn": 3,
    "dropout": 0.1,
    "n_tasks": 4
  },
  "heads": {
    "0": "pEC50 regression (OpenADMET DRC)",
    "1": "NCATS qHTS binary Active/Inactive (AID 1346982)",
    "2": "NCATS qHTS LogAC50 regression (AID 1346985)",
    "3": "Tox21 SR-ARE binary (SR-PXR absent, using proxy)"
  },
  "fold_metrics": [
    {
      "fold": 0,
      "n_train": 3311,
      "n_val": 828,
      "n_train_groups": 2383,
      "n_val_groups": 594,
      "rae": 0.5453029144438072,
      "mae": 0.49125529415365576,
      "r2": 0.6255180288795432,
      "spearman": 0.7447037709863925,
      "kendall": 0.5592024203842534
    },
    {
      "fold": 1,
      "n_train": 3311,
      "n_val": 828,
      "n_train_groups": 2381,
      "n_val_groups": 596,
      "rae": 0.5519226703694166,
      "mae": 0.5112496250896638,
      "r2": 0.6449334144917254,
      "spearman": 0.7385215867617947,
      "kendall": 0.546298762818552
    },
    {
      "fold": 2,
      "n_train": 3311,
      "n_val": 828,
      "n_train_groups": 2381,
      "n_val_groups": 596,
      "rae": 0.5446316449861985,
      "mae": 0.47574268182118734,
      "r2": 0.6556162850975792,
      "spearman": 0.7614076086426662,
      "kendall": 0.5713904175711361
    },
    {
      "fold": 3,
      "n_train": 3311,
      "n_val": 828,
      "n_train_groups": 2381,
      "n_val_groups": 596,
      "rae": 0.5436381961963428,
      "mae": 0.4978908085782816,
      "r2": 0.644040373333556,
      "spearman": 0.7555060410851847,
      "kendall": 0.5641049286538986
    },
    {
      "fold": 4,
      "n_train": 3312,
      "n_val": 827,
      "n_train_groups": 2382,
      "n_val_groups": 595,
      "rae": 0.5266427166244895,
      "mae": 0.49054308552840026,
      "r2": 0.6562898251244359,
      "spearman": 0.7705609628233591,
      "kendall": 0.573085818683251
    }
  ],
  "aggregate": {
    "rae": 0.542228887610097,
    "mae": 0.4933369738864982,
    "r2": 0.6454204917207755,
    "spearman": 0.7540251223132772,
    "kendall": 0.5620542592863466
  },
  "test_preds": {
    "min": 2.582848072052002,
    "max": 6.2093908309936525,
    "mean": 4.884883669431214
  },
  "data_counts": {
    "train_openadmet": 4139,
    "ncats_binary_post_dedup": 8571,
    "ncats_logac50_post_dedup": 644,
    "tox21_post_dedup": 5928,
    "combined_total": 12540
  }
}