checkpoints/metrics_stage1.json

{
  "NR-AhR": {
    "val_auc": 0.8789764868603043,
    "n_train_samples": 8165,
    "n_val_samples": 271
  },
  "NR-AR": {
    "val_auc": 0.8547453703703702,
    "n_train_samples": 9358,
    "n_val_samples": 291
  },
  "NR-AR-LBD": {
    "val_auc": 0.9717741935483871,
    "n_train_samples": 8595,
    "n_val_samples": 252
  },
  "NR-Aromatase": {
    "val_auc": 0.8483560090702948,
    "n_train_samples": 7222,
    "n_val_samples": 214
  },
  "NR-ER": {
    "val_auc": 0.7974683544303797,
    "n_train_samples": 7694,
    "n_val_samples": 264
  },
  "NR-ER-LBD": {
    "val_auc": 0.8347826086956521,
    "n_train_samples": 8749,
    "n_val_samples": 286
  },
  "NR-PPAR-gamma": {
    "val_auc": 0.8077025232403718,
    "n_train_samples": 8180,
    "n_val_samples": 266
  },
  "SR-ARE": {
    "val_auc": 0.8194921070693205,
    "n_train_samples": 7165,
    "n_val_samples": 233
  },
  "SR-ATAD5": {
    "val_auc": 0.8749593495934959,
    "n_train_samples": 9087,
    "n_val_samples": 271
  },
  "SR-HSE": {
    "val_auc": 0.92421875,
    "n_train_samples": 8147,
    "n_val_samples": 266
  },
  "SR-MMP": {
    "val_auc": 0.9280613594287226,
    "n_train_samples": 7317,
    "n_val_samples": 237
  },
  "SR-p53": {
    "val_auc": 0.8038690476190476,
    "n_train_samples": 8630,
    "n_val_samples": 268
  }
}