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

294 lines
7.2 KiB
JSON

{
"fold_results": [
{
"fold_idx": 0,
"n_samples": 87,
"size": {
"n": 87,
"rmse": 0.5578980261231201,
"mae": 0.293820194814397,
"r2": -0.16350852977979646
},
"delivery": {
"n": 64,
"rmse": 1.4180242556680394,
"mae": 0.489678907149937,
"r2": 0.006066588761596381
},
"pdi": {
"n": 87,
"accuracy": 0.7011494252873564,
"precision": 0.382983682983683,
"recall": 0.3981244671781756,
"f1": 0.38831483607603007
},
"ee": {
"n": 87,
"accuracy": 0.6206896551724138,
"precision": 0.4700393567388641,
"recall": 0.5084656084656085,
"f1": 0.4656627032698024
},
"toxic": {
"n": 65,
"accuracy": 0.9846153846153847,
"precision": 0.9918032786885246,
"recall": 0.9,
"f1": 0.9403122130394859
},
"biodist": {
"n": 65,
"kl_divergence": 0.20084619060054956,
"js_divergence": 0.048292698388857934
}
},
{
"fold_idx": 1,
"n_samples": 87,
"size": {
"n": 87,
"rmse": 0.40406802223294774,
"mae": 0.2981993521767101,
"r2": 0.09717283856858072
},
"delivery": {
"n": 58,
"rmse": 0.545857341773909,
"mae": 0.3866841504832023,
"r2": 0.39045172405838346
},
"pdi": {
"n": 87,
"accuracy": 0.8045977011494253,
"precision": 0.49282452707110247,
"recall": 0.45567765567765567,
"f1": 0.4661145617667357
},
"ee": {
"n": 87,
"accuracy": 0.7241379310344828,
"precision": 0.7064586357039188,
"recall": 0.6503496503496503,
"f1": 0.6654761904761903
},
"toxic": {
"n": 59,
"accuracy": 0.9830508474576272,
"precision": 0.9912280701754386,
"recall": 0.8333333333333333,
"f1": 0.8955752212389381
},
"biodist": {
"n": 58,
"kl_divergence": 0.185822128176519,
"js_divergence": 0.049566546350752166
}
},
{
"fold_idx": 2,
"n_samples": 87,
"size": {
"n": 86,
"rmse": 0.5861093745094258,
"mae": 0.35274335949919944,
"r2": -0.3079648452189143
},
"delivery": {
"n": 61,
"rmse": 0.5034529339588798,
"mae": 0.3725305872618175,
"r2": 0.596618413667312
},
"pdi": {
"n": 87,
"accuracy": 0.7701149425287356,
"precision": 0.7266666666666666,
"recall": 0.6349206349206349,
"f1": 0.6497584541062802
},
"ee": {
"n": 87,
"accuracy": 0.5172413793103449,
"precision": 0.43155828639699606,
"recall": 0.40000766812361016,
"f1": 0.3980599647266314
},
"toxic": {
"n": 61,
"accuracy": 1.0,
"precision": 1.0,
"recall": 1.0,
"f1": 1.0
},
"biodist": {
"n": 61,
"kl_divergence": 0.2646404257700098,
"js_divergence": 0.07024299955112
}
},
{
"fold_idx": 3,
"n_samples": 86,
"size": {
"n": 86,
"rmse": 0.32742961478246685,
"mae": 0.25193805472795355,
"r2": -0.09589933555096875
},
"delivery": {
"n": 68,
"rmse": 0.7277366648519259,
"mae": 0.42998586144462664,
"r2": 0.4053674615039361
},
"pdi": {
"n": 86,
"accuracy": 0.7906976744186046,
"precision": 0.7513157894736842,
"recall": 0.6356534090909091,
"f1": 0.6544642857142857
},
"ee": {
"n": 86,
"accuracy": 0.7441860465116279,
"precision": 0.6962905144216474,
"recall": 0.6327243018419488,
"f1": 0.6557768628760061
},
"toxic": {
"n": 68,
"accuracy": 1.0,
"precision": 1.0,
"recall": 1.0,
"f1": 1.0
},
"biodist": {
"n": 68,
"kl_divergence": 0.3411994760293469,
"js_divergence": 0.07812197338009717
}
},
{
"fold_idx": 4,
"n_samples": 87,
"size": {
"n": 86,
"rmse": 0.28795189907825647,
"mae": 0.21156823080639506,
"r2": 0.2077479731717109
},
"delivery": {
"n": 59,
"rmse": 0.8048179107025805,
"mae": 0.5188011898327682,
"r2": 0.24048521206149798
},
"pdi": {
"n": 87,
"accuracy": 0.6896551724137931,
"precision": 0.4101075268817204,
"recall": 0.425,
"f1": 0.4174194267871083
},
"ee": {
"n": 87,
"accuracy": 0.7011494252873564,
"precision": 0.6600529100529101,
"recall": 0.581219806763285,
"f1": 0.5953238953238954
},
"toxic": {
"n": 60,
"accuracy": 0.95,
"precision": 0.8240740740740741,
"recall": 0.8818181818181818,
"f1": 0.8498748957464553
},
"biodist": {
"n": 59,
"kl_divergence": 0.207699088365002,
"js_divergence": 0.05288953180347253
}
}
],
"summary_stats": {
"size": {
"rmse_mean": 0.43269138734524343,
"rmse_std": 0.12005218734930377,
"r2_mean": -0.05249037976187758,
"r2_std": 0.18417364026118202
},
"delivery": {
"rmse_mean": 0.7999778213910669,
"rmse_std": 0.3285507063187239,
"r2_mean": 0.32779788001054516,
"r2_std": 0.19664259417310184
},
"pdi": {
"accuracy_mean": 0.751242983159583,
"accuracy_std": 0.04703609766404652,
"f1_mean": 0.515214312890088,
"f1_std": 0.11451705845950987
},
"ee": {
"accuracy_mean": 0.6614808874632452,
"accuracy_std": 0.08343692429542732,
"f1_mean": 0.5560599233345052,
"f1_std": 0.10638861908930271
},
"toxic": {
"accuracy_mean": 0.9835332464146024,
"accuracy_std": 0.018265761928954075,
"f1_mean": 0.9371524660049758,
"f1_std": 0.05874632007606866
},
"biodist": {
"kl_mean": 0.24004146178828548,
"kl_std": 0.05720155140134621,
"js_mean": 0.05982274989485996,
"js_std": 0.012080103435264247
}
},
"overall": {
"size": {
"n_samples": 432,
"mse": 0.20179930058631468,
"rmse": 0.44922077043065883,
"mae": 0.2817203010673876,
"r2": -0.09923811531698323
},
"delivery": {
"n_samples": 310,
"mse": 0.760202623547859,
"rmse": 0.8718959935381393,
"mae": 0.4398058238289049,
"r2": 0.23486100707044422
},
"pdi": {
"n_samples": 434,
"accuracy": 0.7511520737327189,
"precision": 0.3287852263755878,
"recall": 0.3184060228452752,
"f1": 0.3212571677885814
},
"ee": {
"n_samples": 434,
"accuracy": 0.6612903225806451,
"precision": 0.5836247086247086,
"recall": 0.5469842657342657,
"f1": 0.5597637622559741
},
"toxic": {
"n_samples": 313,
"accuracy": 0.9840255591054313,
"precision": 0.918076923076923,
"recall": 0.8895126612517916,
"f1": 0.9032337847028998
},
"biodist": {
"n_samples": 311,
"kl_divergence": 0.24254521665201,
"js_divergence": 0.06022985409160514
}
}
}