{
  "fold_results": [
    {
      "fold_idx": 0,
      "best_val_loss": 0.866959414306682,
      "best_val_rmse": 0.939569308267399,
      "best_val_r2": 0.24210831047556391,
      "epochs_trained": 21
    },
    {
      "fold_idx": 1,
      "best_val_loss": 0.7120698080343358,
      "best_val_rmse": 0.89035723356781,
      "best_val_r2": 0.1949605134929706,
      "epochs_trained": 15
    },
    {
      "fold_idx": 2,
      "best_val_loss": 0.7995324550767853,
      "best_val_rmse": 0.9427014903075246,
      "best_val_r2": 0.14637384616637905,
      "epochs_trained": 18
    },
    {
      "fold_idx": 3,
      "best_val_loss": 0.7464333962453039,
      "best_val_rmse": 0.8886072948338112,
      "best_val_r2": 0.16404373856430554,
      "epochs_trained": 16
    },
    {
      "fold_idx": 4,
      "best_val_loss": 0.7502263283314047,
      "best_val_rmse": 0.88562010695961,
      "best_val_r2": 0.09823831053676024,
      "epochs_trained": 11
    }
  ],
  "summary": {
    "val_loss_mean": 0.7750442803989024,
    "val_loss_std": 0.053755669741545144,
    "val_rmse_mean": 0.9093710867872311,
    "val_rmse_std": 0.0259985349306361,
    "val_r2_mean": 0.16914494384719586,
    "val_r2_std": 0.048088236270900114
  },
  "config": {
    "d_model": 256,
    "num_heads": 8,
    "n_attn_layers": 4,
    "fusion_strategy": "attention",
    "head_hidden_dim": 128,
    "dropout": 0.1,
    "use_mpnn": true,
    "mpnn_ensemble_paths": [
      "/home/ubuntu/projects/\u4e34\u65f6\u8dd1\u4e0b\u811a\u672c/EXP/lnp_ml/models/mpnn/all_amine_split_for_LiON/cv_0/fold_0/model_0/model.pt",
      "/home/ubuntu/projects/\u4e34\u65f6\u8dd1\u4e0b\u811a\u672c/EXP/lnp_ml/models/mpnn/all_amine_split_for_LiON/cv_1/fold_0/model_0/model.pt",
      "/home/ubuntu/projects/\u4e34\u65f6\u8dd1\u4e0b\u811a\u672c/EXP/lnp_ml/models/mpnn/all_amine_split_for_LiON/cv_2/fold_0/model_0/model.pt",
      "/home/ubuntu/projects/\u4e34\u65f6\u8dd1\u4e0b\u811a\u672c/EXP/lnp_ml/models/mpnn/all_amine_split_for_LiON/cv_3/fold_0/model_0/model.pt",
      "/home/ubuntu/projects/\u4e34\u65f6\u8dd1\u4e0b\u811a\u672c/EXP/lnp_ml/models/mpnn/all_amine_split_for_LiON/cv_4/fold_0/model_0/model.pt"
    ],
    "lr": 0.0001,
    "weight_decay": 1e-05,
    "batch_size": 64,
    "epochs": 50,
    "patience": 10
  }
}