lnp_ml/models/pretrain_cv/test_results.json
2026-01-21 22:57:44 +08:00

63 lines
1.5 KiB
JSON

{
"fold_results": [
{
"fold_idx": 0,
"n_samples": 2037,
"mse": 0.704966583529581,
"rmse": 0.8396228817329724,
"mae": 0.6275912649623819,
"r2": 0.18947693925460807,
"correlation": 0.4894546998045728
},
{
"fold_idx": 1,
"n_samples": 1658,
"mse": 0.9115043728317782,
"rmse": 0.9547273814193129,
"mae": 0.7232097773077567,
"r2": 0.21745691549383905,
"correlation": 0.48734383353129934
},
{
"fold_idx": 2,
"n_samples": 1615,
"mse": 0.8047081385461913,
"rmse": 0.8970552594718963,
"mae": 0.6672816107495062,
"r2": 0.18280256744724777,
"correlation": 0.506874931057286
},
{
"fold_idx": 3,
"n_samples": 1754,
"mse": 0.8592105232097863,
"rmse": 0.926936094458397,
"mae": 0.6835461141701688,
"r2": 0.1746865688538688,
"correlation": 0.4233578915460334
},
{
"fold_idx": 4,
"n_samples": 1520,
"mse": 0.6832138393192553,
"rmse": 0.8265675043934738,
"mae": 0.6324079235422495,
"r2": 0.2766966748218629,
"correlation": 0.5350317708563872
}
],
"summary_stats": {
"rmse_mean": 0.8889818242952104,
"rmse_std": 0.04931538867410365,
"r2_mean": 0.2082239331742853,
"r2_std": 0.03713816024862206
},
"overall": {
"n_samples": 8584,
"mse": 0.7912902048033758,
"rmse": 0.8895449425427452,
"mae": 0.6658137170204776,
"r2": 0.20870978896035208,
"correlation": 0.4812266887089009
}
}