lnp_ml/models/finetune_cv/test_results.json
2026-01-22 17:06:24 +08:00

294 lines
7.1 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,
"precision": 0.20350877192982456,
"recall": 0.3333333333333333,
"f1": 0.25272331154684097
},
"ee": {
"n": 95,
"accuracy": 0.6631578947368421,
"precision": 0.22580645161290322,
"recall": 0.328125,
"f1": 0.267515923566879
},
"toxic": {
"n": 66,
"accuracy": 0.8939393939393939,
"precision": 0.44696969696969696,
"recall": 0.5,
"f1": 0.472
},
"biodist": {
"n": 66,
"kl_divergence": 0.8735123101690443,
"js_divergence": 0.2203766048579219
}
},
{
"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,
"precision": 0.35384615384615387,
"recall": 0.5,
"f1": 0.4144144144144144
},
"ee": {
"n": 195,
"accuracy": 0.4205128205128205,
"precision": 0.14017094017094017,
"recall": 0.3333333333333333,
"f1": 0.19735258724428398
},
"toxic": {
"n": 123,
"accuracy": 1.0,
"precision": 1.0,
"recall": 1.0,
"f1": 1.0
},
"biodist": {
"n": 123,
"kl_divergence": 1.218490162626268,
"js_divergence": 0.3069412988090679
}
},
{
"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,
"precision": 0.29411764705882354,
"recall": 0.3333333333333333,
"f1": 0.3125
},
"ee": {
"n": 51,
"accuracy": 0.8431372549019608,
"precision": 0.28104575163398693,
"recall": 0.3333333333333333,
"f1": 0.3049645390070922
},
"toxic": {
"n": 47,
"accuracy": 0.851063829787234,
"precision": 0.425531914893617,
"recall": 0.5,
"f1": 0.4597701149425288
},
"biodist": {
"n": 45,
"kl_divergence": 1.0902737810034395,
"js_divergence": 0.27372590260311874
}
},
{
"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,
"precision": 0.42424242424242425,
"recall": 0.5,
"f1": 0.4590163934426229
},
"ee": {
"n": 66,
"accuracy": 0.18181818181818182,
"precision": 0.203921568627451,
"recall": 0.2707070707070707,
"f1": 0.12297410192147035
},
"toxic": {
"n": 62,
"accuracy": 1.0,
"precision": 1.0,
"recall": 1.0,
"f1": 1.0
},
"biodist": {
"n": 62,
"kl_divergence": 0.9434472801013084,
"js_divergence": 0.20391570021898991
}
},
{
"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,
"precision": 0.4444444444444444,
"recall": 0.5,
"f1": 0.47058823529411764
},
"ee": {
"n": 27,
"accuracy": 0.5925925925925926,
"precision": 0.19753086419753085,
"recall": 0.3333333333333333,
"f1": 0.24806201550387597
},
"toxic": {
"n": 15,
"accuracy": 1.0,
"precision": 1.0,
"recall": 1.0,
"f1": 1.0
},
"biodist": {
"n": 15,
"kl_divergence": 0.7615404990024607,
"js_divergence": 0.20916908426734354
}
}
],
"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,
"f1_mean": 0.3818484709395992,
"f1_std": 0.08529090446864619
},
"ee": {
"accuracy_mean": 0.5402437489124795,
"accuracy_std": 0.22467627690136344,
"f1_mean": 0.2281738334487203,
"f1_std": 0.063019179670124
},
"toxic": {
"accuracy_mean": 0.9490006447453256,
"accuracy_std": 0.06391582554207781,
"f1_mean": 0.7863540229885058,
"f1_std": 0.26169039387919035
},
"biodist": {
"kl_mean": 0.9774528065805042,
"kl_std": 0.1608761675751487,
"js_mean": 0.24282571815128842,
"js_std": 0.04053726747029075
}
},
"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,
"precision": 0.18490783410138248,
"recall": 0.25,
"f1": 0.21258278145695364
},
"ee": {
"n_samples": 434,
"accuracy": 0.4976958525345622,
"precision": 0.21063404810247777,
"recall": 0.2839160839160839,
"f1": 0.23684873775319115
},
"toxic": {
"n_samples": 313,
"accuracy": 0.9552715654952076,
"precision": 0.4776357827476038,
"recall": 0.5,
"f1": 0.48856209150326796
},
"biodist": {
"n_samples": 311,
"kl_divergence": 1.0498561462079121,
"js_divergence": 0.25851000311532496
}
}
}