mirror of
https://github.com/RYDE-WORK/lnp_ml.git
synced 2026-03-21 01:27:00 +08:00
68 lines
2.2 KiB
JSON
68 lines
2.2 KiB
JSON
{
|
|
"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
|
|
}
|
|
} |