lnp_ml/models/benchmark/test_results.json
2026-03-02 14:51:54 +08:00

63 lines
1.5 KiB
JSON

{
"fold_results": [
{
"fold_idx": 0,
"n_samples": 2037,
"mse": 0.7226313117346976,
"rmse": 0.8500772386875781,
"mae": 0.6242580430768412,
"r2": 0.16916722542340512,
"correlation": 0.4730393645703907
},
{
"fold_idx": 1,
"n_samples": 1658,
"mse": 0.9158973512242636,
"rmse": 0.9570252615392467,
"mae": 0.7200029971805739,
"r2": 0.21368546363481555,
"correlation": 0.4973138359369983
},
{
"fold_idx": 2,
"n_samples": 1615,
"mse": 0.7387503133202947,
"rmse": 0.8595058541512645,
"mae": 0.6402343517433183,
"r2": 0.24978407645589784,
"correlation": 0.5249250503279256
},
{
"fold_idx": 3,
"n_samples": 1754,
"mse": 0.9009732149716101,
"rmse": 0.949196088788618,
"mae": 0.70815324582605,
"r2": 0.134571475403793,
"correlation": 0.4336201644807484
},
{
"fold_idx": 4,
"n_samples": 1520,
"mse": 0.7580491877309711,
"rmse": 0.870660202220689,
"mae": 0.653891459692654,
"r2": 0.1974701527113163,
"correlation": 0.488930703580102
}
],
"summary_stats": {
"rmse_mean": 0.8972929290774792,
"rmse_std": 0.046105045536526425,
"r2_mean": 0.19293567872584555,
"r2_std": 0.03912588645481614
},
"overall": {
"n_samples": 8584,
"mse": 0.8057061196128802,
"rmse": 0.8976113410674356,
"mae": 0.6681468887523418,
"r2": 0.19429387403723375,
"correlation": 0.4821095149481418
}
}