lnp_ml/models/finetune_cv/test_results.json
RYDE-WORK 039be54c5a ...
2026-01-22 01:01:29 +08:00

198 lines
4.2 KiB
JSON

{
"fold_results": [
{
"fold_idx": 0,
"n_samples": 95,
"size": {
"n": 95,
"rmse": 0.632240295394253,
"mae": 0.4674149764211554,
"r2": -0.13819694158244777
},
"delivery": {
"n": 66,
"rmse": 1.3553486689823926,
"mae": 0.48114709207562334,
"r2": -0.008234200686471516
},
"pdi": {
"n": 95,
"accuracy": 0.6105263157894737
},
"ee": {
"n": 95,
"accuracy": 0.6631578947368421
},
"toxic": {
"n": 66,
"accuracy": 0.8939393939393939
}
},
{
"fold_idx": 1,
"n_samples": 195,
"size": {
"n": 193,
"rmse": 0.42622752144538556,
"mae": 0.24566329575573226,
"r2": 0.0482086242002292
},
"delivery": {
"n": 123,
"rmse": 0.742899240869997,
"mae": 0.5315999669170507,
"r2": -0.03140039086191759
},
"pdi": {
"n": 195,
"accuracy": 0.7076923076923077
},
"ee": {
"n": 195,
"accuracy": 0.4205128205128205
},
"toxic": {
"n": 123,
"accuracy": 1.0
}
},
{
"fold_idx": 2,
"n_samples": 51,
"size": {
"n": 51,
"rmse": 0.241909571406037,
"mae": 0.20043573192521638,
"r2": -0.43487628292073
},
"delivery": {
"n": 44,
"rmse": 0.7564153649581582,
"mae": 0.6047130756302398,
"r2": -0.4226486727361405
},
"pdi": {
"n": 51,
"accuracy": 0.8823529411764706
},
"ee": {
"n": 51,
"accuracy": 0.8431372549019608
},
"toxic": {
"n": 47,
"accuracy": 0.851063829787234
}
},
{
"fold_idx": 3,
"n_samples": 66,
"size": {
"n": 66,
"rmse": 0.2857872773679936,
"mae": 0.22075237649859805,
"r2": -0.5674047032859011
},
"delivery": {
"n": 62,
"rmse": 1.0291312965402932,
"mae": 0.7422042032328224,
"r2": -0.7148264932933832
},
"pdi": {
"n": 66,
"accuracy": 0.8484848484848485
},
"ee": {
"n": 66,
"accuracy": 0.18181818181818182
},
"toxic": {
"n": 62,
"accuracy": 1.0
}
},
{
"fold_idx": 4,
"n_samples": 27,
"size": {
"n": 27,
"rmse": 0.2271495001169846,
"mae": 0.18753767013549805,
"r2": -0.19441156195074893
},
"delivery": {
"n": 15,
"rmse": 1.993006453768918,
"mae": 1.3779302000999452,
"r2": -0.3411461507368889
},
"pdi": {
"n": 27,
"accuracy": 0.8888888888888888
},
"ee": {
"n": 27,
"accuracy": 0.5925925925925926
},
"toxic": {
"n": 15,
"accuracy": 1.0
}
}
],
"summary_stats": {
"size": {
"rmse_mean": 0.36266283314613074,
"rmse_std": 0.15203127472757474,
"r2_mean": -0.2573361731079197,
"r2_std": 0.2187118059634264
},
"delivery": {
"rmse_mean": 1.1753602050239518,
"rmse_std": 0.46580283242073095,
"r2_mean": -0.30365118166296035,
"r2_std": 0.2630677092396549
},
"pdi": {
"accuracy_mean": 0.7875890604063979,
"accuracy_std": 0.11016791908756088
},
"ee": {
"accuracy_mean": 0.5402437489124795,
"accuracy_std": 0.22467627690136344
},
"toxic": {
"accuracy_mean": 0.9490006447453256,
"accuracy_std": 0.06391582554207781
}
},
"overall": {
"size": {
"n_samples": 432,
"mse": 0.19167728610863985,
"rmse": 0.43780964597486866,
"mae": 0.2816500812768936,
"r2": -0.04410163027802061
},
"delivery": {
"n_samples": 310,
"mse": 1.095306046771274,
"rmse": 1.0465687014101244,
"mae": 0.6143080417337197,
"r2": -0.1024184074306409
},
"pdi": {
"n_samples": 434,
"accuracy": 0.7396313364055299
},
"ee": {
"n_samples": 434,
"accuracy": 0.4976958525345622
},
"toxic": {
"n_samples": 313,
"accuracy": 0.9552715654952076
}
}
}