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

294 lines
7.1 KiB
JSON

{
"fold_results": [
{
"fold_idx": 0,
"n_samples": 95,
"size": {
"n": 95,
"rmse": 0.5909209144067168,
"mae": 0.376253614927593,
"r2": 0.005712927161997228
},
"delivery": {
"n": 66,
"rmse": 1.3280577883458438,
"mae": 0.5195405159964028,
"r2": 0.03195999739694366
},
"pdi": {
"n": 95,
"accuracy": 0.6105263157894737,
"precision": 0.20350877192982456,
"recall": 0.3333333333333333,
"f1": 0.25272331154684097
},
"ee": {
"n": 95,
"accuracy": 0.6736842105263158,
"precision": 0.22456140350877193,
"recall": 0.3333333333333333,
"f1": 0.26834381551362685
},
"toxic": {
"n": 66,
"accuracy": 0.8939393939393939,
"precision": 0.44696969696969696,
"recall": 0.5,
"f1": 0.472
},
"biodist": {
"n": 66,
"kl_divergence": 0.851655784204727,
"js_divergence": 0.21404831573756974
}
},
{
"fold_idx": 1,
"n_samples": 195,
"size": {
"n": 193,
"rmse": 0.4425801645813746,
"mae": 0.26432527161632796,
"r2": -0.026225211870033682
},
"delivery": {
"n": 123,
"rmse": 0.7771322048436382,
"mae": 0.6133777339870822,
"r2": -0.128644776760948
},
"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": 0.9336461102028436,
"js_divergence": 0.24870266224462317
}
},
{
"fold_idx": 2,
"n_samples": 51,
"size": {
"n": 51,
"rmse": 0.6473513298834871,
"mae": 0.5600235602434944,
"r2": -9.27515642706235
},
"delivery": {
"n": 44,
"rmse": 0.7721077356414991,
"mae": 0.6167582499593581,
"r2": -0.4822886602727561
},
"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.1049896129018548,
"js_divergence": 0.25485248115851133
}
},
{
"fold_idx": 3,
"n_samples": 66,
"size": {
"n": 66,
"rmse": 0.2407212117920812,
"mae": 0.19363613562150436,
"r2": -0.11204941379936861
},
"delivery": {
"n": 62,
"rmse": 1.0041711455927012,
"mae": 0.7132550483914993,
"r2": -0.63265374674746
},
"pdi": {
"n": 66,
"accuracy": 0.8484848484848485,
"precision": 0.42424242424242425,
"recall": 0.5,
"f1": 0.4590163934426229
},
"ee": {
"n": 66,
"accuracy": 0.8181818181818182,
"precision": 0.27692307692307694,
"recall": 0.32727272727272727,
"f1": 0.3
},
"toxic": {
"n": 62,
"accuracy": 1.0,
"precision": 1.0,
"recall": 1.0,
"f1": 1.0
},
"biodist": {
"n": 62,
"kl_divergence": 0.9677978984139058,
"js_divergence": 0.2020309307244639
}
},
{
"fold_idx": 4,
"n_samples": 27,
"size": {
"n": 27,
"rmse": 0.23392834445509142,
"mae": 0.19066280788845485,
"r2": -0.2667651950955112
},
"delivery": {
"n": 15,
"rmse": 1.9603892288630869,
"mae": 1.3892907698949177,
"r2": -0.29760739742916287
},
"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.9389607012315264,
"js_divergence": 0.2470218476598176
}
}
],
"summary_stats": {
"size": {
"rmse_mean": 0.43110039302375025,
"rmse_std": 0.17179051271013462,
"r2_mean": -1.9348966641330534,
"r2_std": 3.6713441784129
},
"delivery": {
"rmse_mean": 1.1683716206573538,
"rmse_std": 0.4449374578352648,
"r2_mean": -0.30184691676267666,
"r2_std": 0.23809090378746706
},
"pdi": {
"accuracy_mean": 0.7875890604063979,
"accuracy_std": 0.11016791908756088,
"f1_mean": 0.3818484709395992,
"f1_std": 0.08529090446864619
},
"ee": {
"accuracy_mean": 0.6696217393431015,
"accuracy_std": 0.15503740047242787,
"f1_mean": 0.2637445914537758,
"f1_std": 0.039213602228007696
},
"toxic": {
"accuracy_mean": 0.9490006447453256,
"accuracy_std": 0.06391582554207781,
"f1_mean": 0.7863540229885058,
"f1_std": 0.26169039387919035
},
"biodist": {
"kl_mean": 0.9594100213909715,
"kl_std": 0.08240959093662605,
"js_mean": 0.23333124750499712,
"js_std": 0.021158533549255752
}
},
"overall": {
"size": {
"n_samples": 432,
"mse": 0.22604480336185886,
"rmse": 0.47544169291497657,
"mae": 0.3084443360567093,
"r2": -0.2313078534105617
},
"delivery": {
"n_samples": 310,
"mse": 1.0873755440675295,
"rmse": 1.0427730069710903,
"mae": 0.6513989447841361,
"r2": -0.09443640807387799
},
"pdi": {
"n_samples": 434,
"accuracy": 0.7396313364055299,
"precision": 0.18490783410138248,
"recall": 0.25,
"f1": 0.21258278145695364
},
"ee": {
"n_samples": 434,
"accuracy": 0.5967741935483871,
"precision": 0.1993841416474211,
"recall": 0.33205128205128204,
"f1": 0.24915824915824913
},
"toxic": {
"n_samples": 313,
"accuracy": 0.9552715654952076,
"precision": 0.4776357827476038,
"recall": 0.5,
"f1": 0.48856209150326796
},
"biodist": {
"n_samples": 311,
"kl_divergence": 0.9481034280166569,
"js_divergence": 0.23285280825310384
}
}
}