RYDE-WORK/lnp_ml
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/lnp_ml.git synced 2026-07-22 05:49:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add 6-variant ablation results (baseline/moe/llm_molt5/llm_biot5/both_molt5/both_biot5)

Browse Source 
- Updated ablation_summary.csv with full 6-variant comparison
- Updated run_all.sh to run 6 variants (added BioT5 variants)
- Added per-fold json artifacts (best_params/test_metrics/splits/history/summary)
- Model weights (.pt) excluded per .gitignore
This commit is contained in:
Michelle0474 2026-06-24 15:33:07 +08:00
parent 182f7853d7
commit c0aa30f2bb
464 changed files with 44287 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
models/abl/baseline/20260623_205158/outer_fold_0/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.18569939775458322,
"lr": 0.0009872295492725695,
"weight_decay": 0.017393516831905452,
"backbone_lr_ratio": 0.17239541174804215,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/outer_fold_0/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 11}

235
models/abl/baseline/20260623_205158/outer_fold_0/history.json Normal file
View File

@ -0,0 +1,235 @@
{
"train": [
{
"loss": 5.497621965408325,
"loss_size": 0.8539485037326813,
"loss_pdi": 0.6807325124740601,
"loss_ee": 1.074701851606369,
"loss_delivery": 1.1345060855150222,
"loss_biodist": 1.1201433956623077,
"loss_toxic": 0.640286660194397
},
{
"loss": 4.665785837173462,
"loss_size": 0.7771127209067344,
"loss_pdi": 0.6143146187067032,
"loss_ee": 0.9874029815196991,
"loss_delivery": 1.0855722874403,
"loss_biodist": 0.7573332965373993,
"loss_toxic": 0.4594721615314484
},
{
"loss": 4.055382227897644,
"loss_size": 0.7433859005570411,
"loss_pdi": 0.5953219294548034,
"loss_ee": 0.9137985289096833,
"loss_delivery": 0.8721775218844414,
"loss_biodist": 0.5785491526126861,
"loss_toxic": 0.3676828071475029
},
{
"loss": 3.6321866273880006,
"loss_size": 0.717304015159607,
"loss_pdi": 0.533730635046959,
"loss_ee": 0.8438478291034699,
"loss_delivery": 0.8553364604711533,
"loss_biodist": 0.46030169576406477,
"loss_toxic": 0.23670255541801452
},
{
"loss": 3.1775063037872315,
"loss_size": 0.6448001429438591,
"loss_pdi": 0.48881121575832365,
"loss_ee": 0.7834544956684113,
"loss_delivery": 0.7532066002488136,
"loss_biodist": 0.37827192842960355,
"loss_toxic": 0.14029031172394751
},
{
"loss": 3.1324957370758058,
"loss_size": 0.60723547488451,
"loss_pdi": 0.479318767786026,
"loss_ee": 0.7696647226810456,
"loss_delivery": 0.8410048380494117,
"loss_biodist": 0.33535364717245103,
"loss_toxic": 0.11438785009086132
},
{
"loss": 2.8942618370056152,
"loss_size": 0.5271590426564217,
"loss_pdi": 0.47021051347255705,
"loss_ee": 0.7096212565898895,
"loss_delivery": 0.7879070624709129,
"loss_biodist": 0.30485892444849017,
"loss_toxic": 0.10716681871563197
},
{
"loss": 2.7531323194503785,
"loss_size": 0.4663904130458832,
"loss_pdi": 0.47090265452861785,
"loss_ee": 0.715871149301529,
"loss_delivery": 0.6963218525052071,
"loss_biodist": 0.3017851024866104,
"loss_toxic": 0.11408571600914001
},
{
"loss": 2.6009469985961915,
"loss_size": 0.46345452666282655,
"loss_pdi": 0.4472439229488373,
"loss_ee": 0.700391435623169,
"loss_delivery": 0.6605987250804901,
"loss_biodist": 0.2730583265423775,
"loss_toxic": 0.06781168337911367
},
{
"loss": 2.5882774353027345,
"loss_size": 0.47313612550497053,
"loss_pdi": 0.4456209808588028,
"loss_ee": 0.6821118712425231,
"loss_delivery": 0.6589170470833778,
"loss_biodist": 0.2672285035252571,
"loss_toxic": 0.07272858265787363
},
{
"loss": 2.7274032115936278,
"loss_size": 0.4289181873202324,
"loss_pdi": 0.4558003395795822,
"loss_ee": 0.6840484797954559,
"loss_delivery": 0.7679663375020027,
"loss_biodist": 0.2954824432730675,
"loss_toxic": 0.1097218444570899
}
],
"val": [
{
"loss": 109.0036735534668,
"loss_size": 38.92370414733887,
"loss_pdi": 1.4176360500277951,
"loss_ee": 10.048195362091064,
"loss_delivery": 44.40848922729492,
"loss_biodist": 14.602405309677124,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.09090909090909091,
"acc_toxic": 1.0
},
{
"loss": 56.829524993896484,
"loss_size": 19.320612907409668,
"loss_pdi": 2.3153170935911476,
"loss_ee": 7.681936502456665,
"loss_delivery": 13.263970851898193,
"loss_biodist": 14.794893980026245,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.09090909090909091,
"acc_toxic": 1.0
},
{
"loss": 26.599002838134766,
"loss_size": 5.124399423599243,
"loss_pdi": 1.6130303277168423,
"loss_ee": 3.832101345062256,
"loss_delivery": 7.275281667709351,
"loss_biodist": 9.16922378540039,
"loss_toxic": 3.199827318667303e-07,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.09090909090909091,
"acc_toxic": 1.0
},
{
"loss": 6.303462743759155,
"loss_size": 1.0172262787818909,
"loss_pdi": 1.0628789365291595,
"loss_ee": 1.515501394867897,
"loss_delivery": 1.548414558172226,
"loss_biodist": 1.0026478469371796,
"loss_toxic": 0.23662380035966635,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 3.7462174892425537,
"loss_size": 0.3320678352029063,
"loss_pdi": 1.116708666086197,
"loss_ee": 0.6200230866670609,
"loss_delivery": 0.5071754567325115,
"loss_biodist": 0.24906498193740845,
"loss_toxic": 0.9348037466406822,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.9
},
{
"loss": 3.948453187942505,
"loss_size": 0.30042218789458275,
"loss_pdi": 1.490148901939392,
"loss_ee": 0.6006122156977654,
"loss_delivery": 0.5205002576112747,
"loss_biodist": 0.2445919942110777,
"loss_toxic": 0.8150980398058891,
"acc_pdi": 0.6363636363636364,
"acc_ee": 0.3939393939393939,
"acc_toxic": 0.9
},
{
"loss": 3.8061407804489136,
"loss_size": 0.3417436182498932,
"loss_pdi": 1.3472748100757599,
"loss_ee": 0.567518962547183,
"loss_delivery": 0.38223914708942175,
"loss_biodist": 0.25772084202617407,
"loss_toxic": 0.9215792268514633,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.9
},
{
"loss": 3.619380831718445,
"loss_size": 0.2981780245900154,
"loss_pdi": 1.3350921273231506,
"loss_ee": 0.5703712888062,
"loss_delivery": 0.3662942796945572,
"loss_biodist": 0.2550168912857771,
"loss_toxic": 0.806022871285677,
"acc_pdi": 0.6060606060606061,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 3.5948950052261353,
"loss_size": 0.2957660034298897,
"loss_pdi": 1.2681505978107452,
"loss_ee": 0.5546197704970837,
"loss_delivery": 0.4010595977306366,
"loss_biodist": 0.24939686432480812,
"loss_toxic": 0.8345895074307919,
"acc_pdi": 0.6060606060606061,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 3.569096803665161,
"loss_size": 0.28623853623867035,
"loss_pdi": 1.2415551543235779,
"loss_ee": 0.54526112601161,
"loss_delivery": 0.39964332804083824,
"loss_biodist": 0.2504266854375601,
"loss_toxic": 0.8528216406702995,
"acc_pdi": 0.6060606060606061,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 3.592081904411316,
"loss_size": 0.2857363075017929,
"loss_pdi": 1.2804671227931976,
"loss_ee": 0.5549933146685362,
"loss_delivery": 0.4104979485273361,
"loss_biodist": 0.2530668321996927,
"loss_toxic": 0.8164789229631424,
"acc_pdi": 0.5454545454545454,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
}
]
}

1
models/abl/baseline/20260623_205158/outer_fold_0/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 21, 22, 23, 24, 25, 26, 27, 28, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 52, 53, 54, 55, 56, 57, 58, 61, 62, 63, 64, 65, 66, 67, 68, 71, 72, 73, 75, 76, 77, 80, 81, 85, 86, 88, 89, 90, 92, 94, 95, 96, 98, 99, 100, 101, 103, 104, 105, 106, 107, 108, 109, 111, 113, 114, 115, 116, 117, 118, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 133, 134, 135, 136, 138, 139, 141, 144, 145, 146, 147, 148, 149, 150, 151, 153, 154, 155, 156, 157, 158, 159, 160, 163, 164, 167, 168, 169, 170, 171, 172, 175, 176, 178, 179, 180, 182, 183, 184, 185, 188, 189, 190, 191, 192, 193, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 206, 208, 210, 211, 212, 213, 216, 217, 218, 219, 220, 222, 224, 225, 227, 228, 229, 230, 231, 232, 233, 234, 236, 237, 239, 241, 242, 244, 245, 247, 248, 249, 250, 252, 253, 255, 256, 257, 258, 259, 260, 261, 262, 265, 267, 268, 269, 270, 271, 272, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 285, 286, 287, 288, 291, 292, 294, 296, 297, 298, 299, 300, 301, 303, 304, 306, 307, 308, 309, 310, 311, 312, 313, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 332, 333, 335, 336, 337, 338, 339, 340, 342, 343, 344, 345, 346, 347, 348, 349, 351, 352, 354, 355, 356, 357, 358, 359, 360, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 375, 377, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 393, 395, 396, 397, 398, 399, 400, 401, 403, 404, 405, 406, 407, 408, 409, 411, 412, 413, 414, 415, 417, 418, 419], "outer_test_idx": [7, 16, 20, 29, 51, 59, 60, 69, 70, 74, 78, 79, 82, 83, 84, 87, 91, 93, 97, 102, 110, 112, 119, 120, 121, 132, 137, 140, 142, 143, 152, 161, 162, 165, 166, 173, 174, 177, 181, 186, 187, 194, 205, 207, 209, 214, 215, 221, 223, 226, 235, 238, 240, 243, 246, 251, 254, 263, 264, 266, 273, 284, 289, 290, 293, 295, 302, 305, 314, 315, 331, 334, 341, 350, 353, 361, 373, 376, 378, 392, 394, 402, 410, 416]}

42
models/abl/baseline/20260623_205158/outer_fold_0/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 83,
"mse": 1.5282298486963632,
"rmse": 1.2362159393473144,
"mae": 0.5704747740717897,
"r2": 0.06468571536968037
},
"delivery": {
"n_samples": 58,
"mse": 0.5436139978785772,
"rmse": 0.7373018363455887,
"mae": 0.5591465608570082,
"r2": 0.3090338048982032
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7738095238095238,
"precision": 0.7232142857142857,
"recall": 0.7494654312188168,
"f1": 0.7325289089994973
},
"ee": {
"n_samples": 84,
"accuracy": 0.7023809523809523,
"precision": 0.6590590922487474,
"recall": 0.6976190476190475,
"f1": 0.666253917297256
},
"toxic": {
"n_samples": 58,
"accuracy": 0.9655172413793104,
"precision": 0.75,
"recall": 0.9821428571428572,
"f1": 0.8242424242424242
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.33174143107041487,
"js_divergence": 0.07276002005717812
}
}

12
models/abl/baseline/20260623_205158/outer_fold_1/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.36640960296080094,
"lr": 0.0009935032389272061,
"weight_decay": 0.007506834667713186,
"backbone_lr_ratio": 0.10108281143505408,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/outer_fold_1/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 11}

237
models/abl/baseline/20260623_205158/outer_fold_1/history.json Normal file
View File

@ -0,0 +1,237 @@
{
"train": [
{
"loss": 5.945608711242675,
"loss_size": 1.5304708689451219,
"loss_pdi": 0.6726944208145141,
"loss_ee": 1.0644152641296387,
"loss_delivery": 0.8849235534667969,
"loss_biodist": 1.1411399304866792,
"loss_toxic": 0.6605329066514969
},
{
"loss": 4.949713110923767,
"loss_size": 1.1195905804634094,
"loss_pdi": 0.6338227927684784,
"loss_ee": 0.9838751316070556,
"loss_delivery": 0.8648343622684479,
"loss_biodist": 0.7841578483581543,
"loss_toxic": 0.5816383093595505
},
{
"loss": 4.467831039428711,
"loss_size": 1.0727383553981782,
"loss_pdi": 0.5936303436756134,
"loss_ee": 0.9442179501056671,
"loss_delivery": 0.7905248045921326,
"loss_biodist": 0.6185098588466644,
"loss_toxic": 0.47035433650016784
},
{
"loss": 4.18380217552185,
"loss_size": 1.044257727265358,
"loss_pdi": 0.5976608961820602,
"loss_ee": 0.8851674854755401,
"loss_delivery": 0.8273541167378425,
"loss_biodist": 0.4965231567621231,
"loss_toxic": 0.3555662304162979
},
{
"loss": 3.929092860221863,
"loss_size": 0.9553224802017212,
"loss_pdi": 0.548481023311615,
"loss_ee": 0.8596946358680725,
"loss_delivery": 0.856061440706253,
"loss_biodist": 0.4580619215965271,
"loss_toxic": 0.27367697954177855
},
{
"loss": 3.68485951423645,
"loss_size": 0.9775789946317672,
"loss_pdi": 0.5335725724697113,
"loss_ee": 0.8395525097846985,
"loss_delivery": 0.7799048483371734,
"loss_biodist": 0.3703513115644455,
"loss_toxic": 0.20359595194458963
},
{
"loss": 3.5928884506225587,
"loss_size": 1.0007280588150025,
"loss_pdi": 0.5454913228750229,
"loss_ee": 0.7852838158607482,
"loss_delivery": 0.7570646733045578,
"loss_biodist": 0.34295395016670227,
"loss_toxic": 0.18110380321741104
},
{
"loss": 3.4778535842895506,
"loss_size": 0.9451205372810364,
"loss_pdi": 0.4993386536836624,
"loss_ee": 0.7792195081710815,
"loss_delivery": 0.72136792242527,
"loss_biodist": 0.35201560258865355,
"loss_toxic": 0.19911173582077027
},
{
"loss": 3.327615189552307,
"loss_size": 0.8601643905043602,
"loss_pdi": 0.48963019251823425,
"loss_ee": 0.7834869205951691,
"loss_delivery": 0.7026658147573471,
"loss_biodist": 0.3422913134098053,
"loss_toxic": 0.16675359681248664
},
{
"loss": 3.463587927818298,
"loss_size": 0.9787588834762573,
"loss_pdi": 0.5153584718704224,
"loss_ee": 0.7670513689517975,
"loss_delivery": 0.7177590146660805,
"loss_biodist": 0.34108778089284897,
"loss_toxic": 0.1642155833542347
},
{
"loss": 3.5189871549606324,
"loss_size": 1.0686501011252403,
"loss_pdi": 0.4941871464252472,
"loss_ee": 0.7697622537612915,
"loss_delivery": 0.7022925913333893,
"loss_biodist": 0.3288215175271034,
"loss_toxic": 0.17648260444402694
}
],
"val": [
{
"loss": 40.87331962585449,
"loss_size": 5.631967067718506,
"loss_pdi": 6.588121294975281,
"loss_ee": 9.879969120025635,
"loss_delivery": 9.864555835723877,
"loss_biodist": 9.211999654769897,
"loss_toxic": 3.0099883588263765e-05,
"acc_pdi": 0.36363636363636365,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 30.792519569396973,
"loss_size": 10.967554569244385,
"loss_pdi": 2.806207239627838,
"loss_ee": 8.072541952133179,
"loss_delivery": 0.825436919927597,
"loss_biodist": 8.507388591766357,
"loss_toxic": 0.00010917877807514742,
"acc_pdi": 0.36363636363636365,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 16.113176345825195,
"loss_size": 3.9671990871429443,
"loss_pdi": 1.0375248789787292,
"loss_ee": 5.073045969009399,
"loss_delivery": 0.5164388865232468,
"loss_biodist": 5.7943830490112305,
"loss_toxic": 0.0008842391544021666,
"acc_pdi": 0.36363636363636365,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 6.789994239807129,
"loss_size": 0.5488049313426018,
"loss_pdi": 0.919120579957962,
"loss_ee": 2.375373959541321,
"loss_delivery": 0.2023337334394455,
"loss_biodist": 2.840838134288788,
"loss_toxic": 0.017407117877155542,
"acc_pdi": 0.45454545454545453,
"acc_ee": 0.24242424242424243,
"acc_toxic": 1.0
},
{
"loss": 2.9222121238708496,
"loss_size": 0.3637890890240669,
"loss_pdi": 0.6303471624851227,
"loss_ee": 0.903712123632431,
"loss_delivery": 0.29046083986759186,
"loss_biodist": 0.6711188852787018,
"loss_toxic": 0.07332967035472393,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.4043108224868774,
"loss_size": 0.34663330763578415,
"loss_pdi": 0.5695744454860687,
"loss_ee": 0.7216707468032837,
"loss_delivery": 0.24825536459684372,
"loss_biodist": 0.4548633322119713,
"loss_toxic": 0.0610531959682703,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.7575757575757576,
"acc_toxic": 1.0
},
{
"loss": 2.2514675855636597,
"loss_size": 0.3080187663435936,
"loss_pdi": 0.5474054366350174,
"loss_ee": 0.6590358465909958,
"loss_delivery": 0.23016345500946045,
"loss_biodist": 0.45463068783283234,
"loss_toxic": 0.04730352945625782,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.7575757575757576,
"acc_toxic": 1.0
},
{
"loss": 2.235174238681793,
"loss_size": 0.29733629524707794,
"loss_pdi": 0.5552924275398254,
"loss_ee": 0.6516624391078949,
"loss_delivery": 0.2518923357129097,
"loss_biodist": 0.4390745982527733,
"loss_toxic": 0.036091484129428864,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.7575757575757576,
"acc_toxic": 1.0
},
{
"loss": 2.217436730861664,
"loss_size": 0.29160527884960175,
"loss_pdi": 0.5481906980276108,
"loss_ee": 0.6433911621570587,
"loss_delivery": 0.2560356706380844,
"loss_biodist": 0.4422919377684593,
"loss_toxic": 0.03231595177203417,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.7575757575757576,
"acc_toxic": 1.0
},
{
"loss": 2.176724135875702,
"loss_size": 0.2756724879145622,
"loss_pdi": 0.5539553165435791,
"loss_ee": 0.6436327546834946,
"loss_delivery": 0.23576483875513077,
"loss_biodist": 0.4301540330052376,
"loss_toxic": 0.0334843285381794,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.7575757575757576,
"acc_toxic": 1.0
},
{
"loss": 2.1801823377609253,
"loss_size": 0.275998018682003,
"loss_pdi": 0.5556303262710571,
"loss_ee": 0.643779918551445,
"loss_delivery": 0.23737561702728271,
"loss_biodist": 0.43005073070526123,
"loss_toxic": 0.03347549866884947,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.7575757575757576,
"acc_toxic": 1.0
}
]
}

1
models/abl/baseline/20260623_205158/outer_fold_1/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 3, 5, 6, 7, 10, 11, 12, 13, 15, 16, 18, 19, 20, 21, 24, 26, 27, 29, 30, 31, 32, 33, 35, 36, 38, 39, 40, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 65, 66, 67, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 97, 98, 99, 100, 101, 102, 103, 105, 106, 107, 108, 109, 110, 112, 114, 115, 118, 119, 120, 121, 123, 124, 125, 126, 128, 129, 130, 131, 132, 136, 137, 140, 141, 142, 143, 144, 145, 146, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 161, 162, 164, 165, 166, 167, 168, 169, 171, 172, 173, 174, 175, 176, 177, 178, 179, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 194, 195, 196, 197, 198, 199, 201, 202, 204, 205, 206, 207, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 225, 226, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 240, 242, 243, 244, 246, 247, 248, 250, 251, 252, 254, 255, 256, 258, 259, 260, 261, 262, 263, 264, 266, 268, 269, 270, 271, 273, 274, 275, 276, 277, 278, 279, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 295, 296, 297, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 313, 314, 315, 316, 317, 318, 319, 321, 322, 323, 324, 325, 327, 328, 329, 330, 331, 333, 334, 335, 337, 338, 339, 340, 341, 342, 343, 344, 345, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 361, 362, 364, 365, 366, 367, 368, 369, 373, 374, 375, 376, 378, 379, 380, 381, 382, 384, 387, 388, 389, 391, 392, 393, 394, 396, 397, 398, 399, 402, 403, 404, 405, 406, 408, 409, 410, 411, 412, 413, 414, 416, 417, 419], "outer_test_idx": [1, 2, 4, 8, 9, 14, 17, 22, 23, 25, 28, 34, 37, 41, 43, 45, 63, 64, 68, 85, 96, 104, 111, 113, 116, 117, 122, 127, 133, 134, 135, 138, 139, 147, 160, 163, 170, 180, 191, 192, 193, 200, 203, 208, 224, 227, 230, 239, 241, 245, 249, 253, 257, 265, 267, 272, 280, 294, 298, 299, 309, 312, 320, 326, 332, 336, 346, 347, 360, 363, 370, 371, 372, 377, 383, 385, 386, 390, 395, 400, 401, 407, 415, 418]}

42
models/abl/baseline/20260623_205158/outer_fold_1/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.37057228076561216,
"rmse": 0.6087464831648821,
"mae": 0.4609523837321571,
"r2": -0.13159599860936777
},
"delivery": {
"n_samples": 61,
"mse": 1.082995768972751,
"rmse": 1.0406708264253164,
"mae": 0.6707069295779114,
"r2": 0.18025027489571332
},
"pdi": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.5886939571150097,
"recall": 0.6031746031746031,
"f1": 0.5916666666666667
},
"ee": {
"n_samples": 84,
"accuracy": 0.5595238095238095,
"precision": 0.4957294957294957,
"recall": 0.49526663812378097,
"f1": 0.4845300550726907
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9508196721311475,
"precision": 0.75,
"recall": 0.9741379310344828,
"f1": 0.8200589970501475
},
"biodist": {
"n_samples": 61,
"kl_divergence": 0.37009493729038634,
"js_divergence": 0.09345595384422643
}
}

12
models/abl/baseline/20260623_205158/outer_fold_2/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.318342405407684,
"lr": 0.0008937362022799393,
"weight_decay": 0.06633867694372529,
"backbone_lr_ratio": 0.20906844321970927,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/outer_fold_2/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 10}

215
models/abl/baseline/20260623_205158/outer_fold_2/history.json Normal file
View File

@ -0,0 +1,215 @@
{
"train": [
{
"loss": 5.23044183254242,
"loss_size": 0.8307447999715805,
"loss_pdi": 0.6786107182502746,
"loss_ee": 1.06563658118248,
"loss_delivery": 0.9185347318649292,
"loss_biodist": 1.126531755924225,
"loss_toxic": 0.6161585688591004
},
{
"loss": 4.6045283555984495,
"loss_size": 0.8068280577659607,
"loss_pdi": 0.6305933237075806,
"loss_ee": 0.995470529794693,
"loss_delivery": 0.8626847624778747,
"loss_biodist": 0.8186007857322692,
"loss_toxic": 0.5059347420930862
},
{
"loss": 4.230263257026673,
"loss_size": 0.7773509219288826,
"loss_pdi": 0.603335040807724,
"loss_ee": 0.9686323225498199,
"loss_delivery": 0.8304742515087128,
"loss_biodist": 0.6457460820674896,
"loss_toxic": 0.4231612354516983
},
{
"loss": 4.060935354232788,
"loss_size": 1.0159688472747803,
"loss_pdi": 0.5800193667411804,
"loss_ee": 0.9062654078006744,
"loss_delivery": 0.823286435008049,
"loss_biodist": 0.4784534633159637,
"loss_toxic": 0.2763469934463501
},
{
"loss": 3.3131240606307983,
"loss_size": 0.6769453555345535,
"loss_pdi": 0.5471729725599289,
"loss_ee": 0.8249725461006164,
"loss_delivery": 0.7035827174782753,
"loss_biodist": 0.36984227001667025,
"loss_toxic": 0.20224370453506707
},
{
"loss": 3.2439030170440675,
"loss_size": 0.6550803124904633,
"loss_pdi": 0.5386825948953629,
"loss_ee": 0.8373141527175904,
"loss_delivery": 0.702998834848404,
"loss_biodist": 0.3566574349999428,
"loss_toxic": 0.16687902733683585
},
{
"loss": 2.9375685453414917,
"loss_size": 0.5435681790113449,
"loss_pdi": 0.5022439807653427,
"loss_ee": 0.7970928072929382,
"loss_delivery": 0.6844923168420791,
"loss_biodist": 0.2854831710457802,
"loss_toxic": 0.13518655933439733
},
{
"loss": 3.007694435119629,
"loss_size": 0.5840565845370292,
"loss_pdi": 0.4961481750011444,
"loss_ee": 0.7784465372562408,
"loss_delivery": 0.7300153076648712,
"loss_biodist": 0.3088834911584854,
"loss_toxic": 0.12322076186537742
},
{
"loss": 2.9045401215553284,
"loss_size": 0.5961680084466934,
"loss_pdi": 0.4664625495672226,
"loss_ee": 0.7473427474498748,
"loss_delivery": 0.6797532260417938,
"loss_biodist": 0.2967249765992165,
"loss_toxic": 0.1287788238376379
},
{
"loss": 2.941871428489685,
"loss_size": 0.5580962479114533,
"loss_pdi": 0.49031204283237456,
"loss_ee": 0.7355630338191986,
"loss_delivery": 0.7561295434832573,
"loss_biodist": 0.28379142433404925,
"loss_toxic": 0.13036032486706972
}
],
"val": [
{
"loss": 53.373188972473145,
"loss_size": 5.803760349750519,
"loss_pdi": 4.895272970199585,
"loss_ee": 16.83747386932373,
"loss_delivery": 20.859439849853516,
"loss_biodist": 5.3479838371276855,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.24242424242424243,
"acc_toxic": 1.0
},
{
"loss": 34.110063552856445,
"loss_size": 5.765698194503784,
"loss_pdi": 2.2217907905578613,
"loss_ee": 10.992742538452148,
"loss_delivery": 10.538397789001465,
"loss_biodist": 5.0104660987854,
"loss_toxic": 4.649159279779269e-07,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.24242424242424243,
"acc_toxic": 1.0
},
{
"loss": 20.13915252685547,
"loss_size": 5.945749044418335,
"loss_pdi": 1.494463562965393,
"loss_ee": 6.576669931411743,
"loss_delivery": 3.680877447128296,
"loss_biodist": 2.7534615993499756,
"loss_toxic": 8.027768490137532e-05,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.24242424242424243,
"acc_toxic": 1.0
},
{
"loss": 10.967385292053223,
"loss_size": 4.893803596496582,
"loss_pdi": 1.1066998541355133,
"loss_ee": 2.965715765953064,
"loss_delivery": 1.528667688369751,
"loss_biodist": 0.5877177119255066,
"loss_toxic": 0.037885989993810654,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.30303030303030304,
"acc_toxic": 0.95
},
{
"loss": 6.98528790473938,
"loss_size": 3.840322256088257,
"loss_pdi": 0.7787934243679047,
"loss_ee": 1.3006620407104492,
"loss_delivery": 0.8746585845947266,
"loss_biodist": 0.11663766205310822,
"loss_toxic": 0.14556746184825897,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.754446983337402,
"loss_size": 3.637616515159607,
"loss_pdi": 0.7500059008598328,
"loss_ee": 1.321771651506424,
"loss_delivery": 0.9027717709541321,
"loss_biodist": 0.1231476292014122,
"loss_toxic": 0.0937483087182045,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.5454545454545454,
"acc_toxic": 0.95
},
{
"loss": 6.627461910247803,
"loss_size": 3.6050997972488403,
"loss_pdi": 0.7348843812942505,
"loss_ee": 1.262064516544342,
"loss_delivery": 0.8815833926200867,
"loss_biodist": 0.12287028133869171,
"loss_toxic": 0.09477062523365021,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.711225748062134,
"loss_size": 3.705545663833618,
"loss_pdi": 0.7286391854286194,
"loss_ee": 1.2768357694149017,
"loss_delivery": 0.8598540425300598,
"loss_biodist": 0.12457559257745743,
"loss_toxic": 0.09150844067335129,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.6549413204193115,
"loss_size": 3.6655603647232056,
"loss_pdi": 0.7185163497924805,
"loss_ee": 1.2802763283252716,
"loss_delivery": 0.8583065271377563,
"loss_biodist": 0.12477456778287888,
"loss_toxic": 0.08323758095502853,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.589754343032837,
"loss_size": 3.619962215423584,
"loss_pdi": 0.7239548563957214,
"loss_ee": 1.2706141471862793,
"loss_delivery": 0.8468391299247742,
"loss_biodist": 0.12151344865560532,
"loss_toxic": 0.08179092407226562,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5454545454545454,
"acc_toxic": 0.95
}
]
}

1
models/abl/baseline/20260623_205158/outer_fold_2/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 13, 14, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 32, 33, 34, 36, 37, 38, 39, 40, 41, 43, 44, 45, 47, 48, 49, 50, 51, 55, 56, 57, 58, 59, 60, 61, 63, 64, 65, 68, 69, 70, 72, 74, 76, 78, 79, 80, 82, 83, 84, 85, 86, 87, 88, 91, 93, 94, 95, 96, 97, 100, 101, 102, 103, 104, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 142, 143, 145, 147, 148, 149, 150, 151, 152, 153, 156, 157, 158, 159, 160, 161, 162, 163, 165, 166, 167, 168, 170, 171, 172, 173, 174, 175, 177, 179, 180, 181, 183, 184, 185, 186, 187, 188, 191, 192, 193, 194, 196, 197, 199, 200, 202, 203, 204, 205, 207, 208, 209, 210, 211, 212, 213, 214, 215, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 232, 233, 235, 236, 237, 238, 239, 240, 241, 242, 243, 245, 246, 249, 250, 251, 252, 253, 254, 255, 256, 257, 259, 260, 261, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 280, 281, 282, 283, 284, 285, 287, 289, 290, 293, 294, 295, 296, 297, 298, 299, 300, 302, 303, 305, 307, 308, 309, 310, 311, 312, 313, 314, 315, 317, 318, 320, 321, 322, 323, 324, 326, 327, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 358, 360, 361, 363, 364, 365, 366, 368, 369, 370, 371, 372, 373, 376, 377, 378, 379, 380, 381, 382, 383, 385, 386, 387, 388, 389, 390, 392, 394, 395, 398, 399, 400, 401, 402, 404, 405, 407, 408, 409, 410, 412, 413, 414, 415, 416, 417, 418], "outer_test_idx": [10, 12, 15, 31, 35, 42, 46, 52, 53, 54, 62, 66, 67, 71, 73, 75, 77, 81, 89, 90, 92, 98, 99, 105, 123, 124, 141, 144, 146, 154, 155, 164, 169, 176, 178, 182, 189, 190, 195, 198, 201, 206, 216, 217, 218, 219, 231, 234, 244, 247, 248, 258, 262, 270, 277, 278, 279, 286, 288, 291, 292, 301, 304, 306, 316, 319, 325, 328, 343, 357, 359, 362, 367, 374, 375, 384, 391, 393, 396, 397, 403, 406, 411, 419]}

42
models/abl/baseline/20260623_205158/outer_fold_2/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.365749434655222,
"rmse": 0.6047722171654565,
"mae": 0.47718314868619754,
"r2": 0.2776514013890329
},
"delivery": {
"n_samples": 60,
"mse": 0.668501323406245,
"rmse": 0.817619302246617,
"mae": 0.5434193710253263,
"r2": 0.11010887128905211
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7738095238095238,
"precision": 0.7122033898305085,
"recall": 0.7294721407624634,
"f1": 0.719359943731317
},
"ee": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.6145962732919253,
"recall": 0.6785714285714285,
"f1": 0.6278721278721279
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9344262295081968,
"precision": 0.7142857142857143,
"recall": 0.9655172413793103,
"f1": 0.7821428571428571
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.3336122920652159,
"js_divergence": 0.08430196229187475
}
}

12
models/abl/baseline/20260623_205158/outer_fold_3/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.12612237693012526,
"lr": 0.0009464809649969671,
"weight_decay": 0.07051438027639334,
"backbone_lr_ratio": 0.9343703462702573,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/outer_fold_3/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 18}

384
models/abl/baseline/20260623_205158/outer_fold_3/history.json Normal file
View File

@ -0,0 +1,384 @@
{
"train": [
{
"loss": 4.952912664413452,
"loss_size": 0.5413272649049758,
"loss_pdi": 0.6714449167251587,
"loss_ee": 1.0707621097564697,
"loss_delivery": 0.9023627996444702,
"loss_biodist": 1.1351445853710174,
"loss_toxic": 0.638839203119278
},
{
"loss": 4.024391674995423,
"loss_size": 0.4812195152044296,
"loss_pdi": 0.6145903885364532,
"loss_ee": 0.9591623604297638,
"loss_delivery": 0.8099067315459252,
"loss_biodist": 0.7342722713947296,
"loss_toxic": 0.43739462196826934
},
{
"loss": 3.5546415567398073,
"loss_size": 0.43410811126232146,
"loss_pdi": 0.6050658047199249,
"loss_ee": 0.905722725391388,
"loss_delivery": 0.7855473428964614,
"loss_biodist": 0.548415920138359,
"loss_toxic": 0.2898260593414307
},
{
"loss": 3.8351691246032713,
"loss_size": 0.5987535804510117,
"loss_pdi": 0.6589920341968536,
"loss_ee": 0.9094266653060913,
"loss_delivery": 0.8958727836608886,
"loss_biodist": 0.6058854043483735,
"loss_toxic": 0.19028361439704894
},
{
"loss": 3.107722854614258,
"loss_size": 0.428075709939003,
"loss_pdi": 0.5529564678668976,
"loss_ee": 0.826682847738266,
"loss_delivery": 0.7825898319482804,
"loss_biodist": 0.4181223660707474,
"loss_toxic": 0.11808188958093524
},
{
"loss": 3.327130913734436,
"loss_size": 0.4366114571690559,
"loss_pdi": 0.5397854954004287,
"loss_ee": 0.7983810603618622,
"loss_delivery": 0.8119195133447648,
"loss_biodist": 0.36413702368736267,
"loss_toxic": 0.39388715121895074
},
{
"loss": 2.8726818323135377,
"loss_size": 0.388601616024971,
"loss_pdi": 0.563576129078865,
"loss_ee": 0.7733971238136291,
"loss_delivery": 0.7856569856405258,
"loss_biodist": 0.28811348527669906,
"loss_toxic": 0.09685604320839047
},
{
"loss": 2.555454707145691,
"loss_size": 0.39492437094449995,
"loss_pdi": 0.483505842089653,
"loss_ee": 0.7191343247890473,
"loss_delivery": 0.6204074218869209,
"loss_biodist": 0.24920540899038315,
"loss_toxic": 0.10555795449763536
},
{
"loss": 2.422672188282013,
"loss_size": 0.34975273013114927,
"loss_pdi": 0.458137184381485,
"loss_ee": 0.6772623956203461,
"loss_delivery": 0.6736922666430474,
"loss_biodist": 0.20276646539568902,
"loss_toxic": 0.08230449045076967
},
{
"loss": 2.294202220439911,
"loss_size": 0.33352037966251374,
"loss_pdi": 0.4709230989217758,
"loss_ee": 0.6405366629362106,
"loss_delivery": 0.5893109619617463,
"loss_biodist": 0.19749270528554916,
"loss_toxic": 0.08408116018399596
},
{
"loss": 2.192491376399994,
"loss_size": 0.3003266975283623,
"loss_pdi": 0.4377738177776337,
"loss_ee": 0.6481737494468689,
"loss_delivery": 0.5960441708564759,
"loss_biodist": 0.17706188410520554,
"loss_toxic": 0.05971499737352133
},
{
"loss": 2.109206438064575,
"loss_size": 0.33593337535858153,
"loss_pdi": 0.42546393275260924,
"loss_ee": 0.5919660955667496,
"loss_delivery": 0.5819768279790878,
"loss_biodist": 0.14810880571603774,
"loss_toxic": 0.051910745818167923
},
{
"loss": 2.0158029437065124,
"loss_size": 0.2888984590768814,
"loss_pdi": 0.45580657124519347,
"loss_ee": 0.5797202318906785,
"loss_delivery": 0.49837631583213804,
"loss_biodist": 0.1601906016469002,
"loss_toxic": 0.06020610840059817
},
{
"loss": 1.9307556748390198,
"loss_size": 0.2837902203202248,
"loss_pdi": 0.40886927843093873,
"loss_ee": 0.5343310534954071,
"loss_delivery": 0.5027692660689353,
"loss_biodist": 0.1628089003264904,
"loss_toxic": 0.06413605269044638
},
{
"loss": 1.8900991439819337,
"loss_size": 0.2707607850432396,
"loss_pdi": 0.41065731048583987,
"loss_ee": 0.5383243918418884,
"loss_delivery": 0.49631956964731216,
"loss_biodist": 0.13920489847660064,
"loss_toxic": 0.06274361247196794
},
{
"loss": 1.8641821026802063,
"loss_size": 0.24494659006595612,
"loss_pdi": 0.41452458053827285,
"loss_ee": 0.530466502904892,
"loss_delivery": 0.5103539004921913,
"loss_biodist": 0.13406966477632523,
"loss_toxic": 0.0597837814828381
},
{
"loss": 1.847708523273468,
"loss_size": 0.24917670711874962,
"loss_pdi": 0.38985409736633303,
"loss_ee": 0.52494797706604,
"loss_delivery": 0.5089830808341503,
"loss_biodist": 0.12788926288485528,
"loss_toxic": 0.07539637042209506
},
{
"loss": 1.839604139328003,
"loss_size": 0.2928685933351517,
"loss_pdi": 0.39620599448680877,
"loss_ee": 0.5069311082363128,
"loss_delivery": 0.48787759989500046,
"loss_biodist": 0.12458067238330842,
"loss_toxic": 0.05830812929198146
}
],
"val": [
{
"loss": 127.31632995605469,
"loss_size": 84.8887825012207,
"loss_pdi": 11.211012125015259,
"loss_ee": 4.11315411292162,
"loss_delivery": 21.395748138427734,
"loss_biodist": 5.6470112800598145,
"loss_toxic": 0.11013364046812057,
"acc_pdi": 0.24242424242424243,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 50.89113807678223,
"loss_size": 28.744680404663086,
"loss_pdi": 5.962186336517334,
"loss_ee": 3.7384512992939563,
"loss_delivery": 8.140775680541992,
"loss_biodist": 4.536876201629639,
"loss_toxic": 8.219477604143322e-05,
"acc_pdi": 0.24242424242424243,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 18.151119709014893,
"loss_size": 5.379726767539978,
"loss_pdi": 3.953024387359619,
"loss_ee": 2.1168477265164256,
"loss_delivery": 4.374178886413574,
"loss_biodist": 2.5386672019958496,
"loss_toxic": 0.00014371765428222716,
"acc_pdi": 0.24242424242424243,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 4.412895500659943,
"loss_size": 1.8018976598978043,
"loss_pdi": 0.6881746053695679,
"loss_ee": 0.7351040244102478,
"loss_delivery": 0.7907695174217224,
"loss_biodist": 0.3704509735107422,
"loss_toxic": 0.04331689327955246,
"acc_pdi": 0.6060606060606061,
"acc_ee": 0.45454545454545453,
"acc_toxic": 1.0
},
{
"loss": 3.8720513582229614,
"loss_size": 1.9023129642009735,
"loss_pdi": 0.44200094044208527,
"loss_ee": 0.6251227706670761,
"loss_delivery": 0.6282426118850708,
"loss_biodist": 0.26596248149871826,
"loss_toxic": 0.010372676886618137,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 3.9434673190116882,
"loss_size": 1.6669230610132217,
"loss_pdi": 0.5820382833480835,
"loss_ee": 0.7903740108013153,
"loss_delivery": 0.5094354152679443,
"loss_biodist": 0.1222085952758789,
"loss_toxic": 0.27862122654914856,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.98708838224411,
"loss_size": 1.7421648055315018,
"loss_pdi": 0.6395299732685089,
"loss_ee": 0.7537677586078644,
"loss_delivery": 0.5322040319442749,
"loss_biodist": 0.1685592085123062,
"loss_toxic": 0.15958170592784882,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.389622062444687,
"loss_size": 1.4498111754655838,
"loss_pdi": 0.4079725295305252,
"loss_ee": 0.6552765369415283,
"loss_delivery": 0.5138065814971924,
"loss_biodist": 0.1563863307237625,
"loss_toxic": 0.20305201411247253,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.2872218787670135,
"loss_size": 1.421157069504261,
"loss_pdi": 0.47122623026371,
"loss_ee": 0.6863093078136444,
"loss_delivery": 0.42459091544151306,
"loss_biodist": 0.1194615513086319,
"loss_toxic": 0.1626254767179489,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.3841849267482758,
"loss_size": 1.5734435468912125,
"loss_pdi": 0.3665034994482994,
"loss_ee": 0.666161522269249,
"loss_delivery": 0.5163925886154175,
"loss_biodist": 0.10750683397054672,
"loss_toxic": 0.14729638397693634,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.290373742580414,
"loss_size": 1.4376836642622948,
"loss_pdi": 0.44857414066791534,
"loss_ee": 0.6463653147220612,
"loss_delivery": 0.5389675498008728,
"loss_biodist": 0.10938665270805359,
"loss_toxic": 0.10815674811601639,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.126695692539215,
"loss_size": 1.3906826972961426,
"loss_pdi": 0.33044668287038803,
"loss_ee": 0.6577020436525345,
"loss_delivery": 0.5568065643310547,
"loss_biodist": 0.1052083894610405,
"loss_toxic": 0.0835861936211586,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.1715721487998962,
"loss_size": 1.4715920239686966,
"loss_pdi": 0.3508722335100174,
"loss_ee": 0.6248730719089508,
"loss_delivery": 0.5079544186592102,
"loss_biodist": 0.10165911167860031,
"loss_toxic": 0.10624553263187408,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.206980675458908,
"loss_size": 1.5019824802875519,
"loss_pdi": 0.37384696304798126,
"loss_ee": 0.6274422854185104,
"loss_delivery": 0.5025574564933777,
"loss_biodist": 0.09232369065284729,
"loss_toxic": 0.09992881119251251,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.1217209100723267,
"loss_size": 1.4759190380573273,
"loss_pdi": 0.3564824312925339,
"loss_ee": 0.6257454752922058,
"loss_delivery": 0.5067604184150696,
"loss_biodist": 0.07864180207252502,
"loss_toxic": 0.07079844921827316,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.1306984424591064,
"loss_size": 1.461349979043007,
"loss_pdi": 0.34549926221370697,
"loss_ee": 0.6252316981554031,
"loss_delivery": 0.516960859298706,
"loss_biodist": 0.07962959259748459,
"loss_toxic": 0.09345860779285431,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.1250928938388824,
"loss_size": 1.4517452716827393,
"loss_pdi": 0.33850933611392975,
"loss_ee": 0.6430297046899796,
"loss_delivery": 0.5081745982170105,
"loss_biodist": 0.07858067750930786,
"loss_toxic": 0.09684738516807556,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.11262384057045,
"loss_size": 1.4306410253047943,
"loss_pdi": 0.3418179824948311,
"loss_ee": 0.6444746404886246,
"loss_delivery": 0.5124365091323853,
"loss_biodist": 0.08138877898454666,
"loss_toxic": 0.09462910145521164,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.8695652173913043
}
]
}

1
models/abl/baseline/20260623_205158/outer_fold_3/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 4, 5, 7, 8, 9, 10, 11, 12, 14, 15, 16, 17, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 32, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 45, 46, 47, 51, 52, 53, 54, 55, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 73, 74, 75, 76, 77, 78, 79, 81, 82, 83, 84, 85, 86, 87, 89, 90, 91, 92, 93, 95, 96, 97, 98, 99, 100, 102, 104, 105, 106, 109, 110, 111, 112, 113, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 127, 129, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 146, 147, 148, 150, 151, 152, 153, 154, 155, 156, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 189, 190, 191, 192, 193, 194, 195, 198, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 212, 213, 214, 215, 216, 217, 218, 219, 221, 223, 224, 226, 227, 228, 230, 231, 233, 234, 235, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 251, 252, 253, 254, 255, 256, 257, 258, 259, 262, 263, 264, 265, 266, 267, 268, 269, 270, 272, 273, 274, 275, 276, 277, 278, 279, 280, 282, 284, 286, 288, 289, 290, 291, 292, 293, 294, 295, 297, 298, 299, 301, 302, 304, 305, 306, 307, 308, 309, 310, 312, 314, 315, 316, 317, 319, 320, 321, 325, 326, 327, 328, 331, 332, 334, 336, 340, 341, 343, 344, 346, 347, 348, 350, 352, 353, 357, 359, 360, 361, 362, 363, 364, 366, 367, 368, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 382, 383, 384, 385, 386, 387, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 400, 401, 402, 403, 404, 406, 407, 410, 411, 412, 414, 415, 416, 418, 419], "outer_test_idx": [3, 6, 13, 18, 19, 26, 33, 44, 48, 49, 50, 56, 57, 72, 80, 88, 94, 101, 103, 107, 108, 114, 126, 128, 130, 145, 149, 157, 167, 171, 175, 188, 196, 197, 199, 211, 220, 222, 225, 229, 232, 236, 250, 260, 261, 271, 281, 283, 285, 287, 296, 300, 303, 311, 313, 318, 322, 323, 324, 329, 330, 333, 335, 337, 338, 339, 342, 345, 349, 351, 354, 355, 356, 358, 365, 369, 381, 388, 399, 405, 408, 409, 413, 417]}

42
models/abl/baseline/20260623_205158/outer_fold_3/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 83,
"mse": 1.7046487347114636,
"rmse": 1.3056219723608604,
"mae": 0.5513793248122176,
"r2": 0.03583768297668177
},
"delivery": {
"n_samples": 59,
"mse": 0.7345580643934568,
"rmse": 0.8570636291393171,
"mae": 0.5434487631733892,
"r2": 0.28592228790047713
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7380952380952381,
"precision": 0.6612903225806452,
"recall": 0.6612903225806452,
"f1": 0.6612903225806452
},
"ee": {
"n_samples": 84,
"accuracy": 0.6547619047619048,
"precision": 0.6206349206349207,
"recall": 0.5979889391654097,
"f1": 0.5781687846203974
},
"toxic": {
"n_samples": 60,
"accuracy": 0.9333333333333333,
"precision": 0.7142857142857143,
"recall": 0.9649122807017544,
"f1": 0.7818181818181817
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.2397795651876451,
"js_divergence": 0.0585131523487816
}
}

12
models/abl/baseline/20260623_205158/outer_fold_4/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.313123672812509,
"lr": 0.0009765192272158632,
"weight_decay": 0.0053740240978641905,
"backbone_lr_ratio": 0.13566938254683006,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/outer_fold_4/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 13}

279
models/abl/baseline/20260623_205158/outer_fold_4/history.json Normal file
View File

@ -0,0 +1,279 @@
{
"train": [
{
"loss": 5.741525268554687,
"loss_size": 1.156516370177269,
"loss_pdi": 0.685343861579895,
"loss_ee": 1.0735185623168946,
"loss_delivery": 0.9616645455360413,
"loss_biodist": 1.186050945520401,
"loss_toxic": 0.6878812193870545
},
{
"loss": 4.911237740516663,
"loss_size": 1.0674284636974334,
"loss_pdi": 0.6453235149383545,
"loss_ee": 1.0323128998279572,
"loss_delivery": 0.816954655945301,
"loss_biodist": 0.8427061200141907,
"loss_toxic": 0.526440891623497
},
{
"loss": 4.828473591804505,
"loss_size": 1.4331441923975945,
"loss_pdi": 0.6058421909809113,
"loss_ee": 0.9855840563774109,
"loss_delivery": 0.7852936506271362,
"loss_biodist": 0.6166061341762543,
"loss_toxic": 0.4289455056190491
},
{
"loss": 4.154263138771057,
"loss_size": 1.003881934285164,
"loss_pdi": 0.570314809679985,
"loss_ee": 0.9245084047317504,
"loss_delivery": 0.8588605999946595,
"loss_biodist": 0.5064615607261658,
"loss_toxic": 0.31344225257635117
},
{
"loss": 3.7234285831451417,
"loss_size": 0.9465380579233169,
"loss_pdi": 0.5529910802841187,
"loss_ee": 0.8967042148113251,
"loss_delivery": 0.7372372865676879,
"loss_biodist": 0.39559130370616913,
"loss_toxic": 0.2141057148575783
},
{
"loss": 3.560976815223694,
"loss_size": 0.939998310804367,
"loss_pdi": 0.5548159599304199,
"loss_ee": 0.854036146402359,
"loss_delivery": 0.6668682649731636,
"loss_biodist": 0.4186478316783905,
"loss_toxic": 0.14765850007534026
},
{
"loss": 3.18977153301239,
"loss_size": 0.7886543214321137,
"loss_pdi": 0.5008505374193192,
"loss_ee": 0.7949652433395386,
"loss_delivery": 0.6210900515317916,
"loss_biodist": 0.3619673177599907,
"loss_toxic": 0.1378368753939867
},
{
"loss": 3.1121251583099365,
"loss_size": 0.7432004705071449,
"loss_pdi": 0.5181865096092224,
"loss_ee": 0.7901717007160187,
"loss_delivery": 0.6244697391986846,
"loss_biodist": 0.3297477558255196,
"loss_toxic": 0.12361136078834534
},
{
"loss": 2.974079132080078,
"loss_size": 0.6982291430234909,
"loss_pdi": 0.5072934776544571,
"loss_ee": 0.7648645043373108,
"loss_delivery": 0.5948630377650261,
"loss_biodist": 0.3141175121068954,
"loss_toxic": 0.11092467829585076
},
{
"loss": 3.01216607093811,
"loss_size": 0.7623879760503769,
"loss_pdi": 0.48228113949298856,
"loss_ee": 0.7496436059474945,
"loss_delivery": 0.5967873886227608,
"loss_biodist": 0.3255198061466217,
"loss_toxic": 0.11320235505700112
},
{
"loss": 2.9995705842971803,
"loss_size": 0.7575592502951622,
"loss_pdi": 0.4897093653678894,
"loss_ee": 0.7522820651531219,
"loss_delivery": 0.5961266160011292,
"loss_biodist": 0.3096505835652351,
"loss_toxic": 0.11278742104768753
},
{
"loss": 2.8911011934280397,
"loss_size": 0.6771724820137024,
"loss_pdi": 0.49721924364566805,
"loss_ee": 0.7677179098129272,
"loss_delivery": 0.572163051366806,
"loss_biodist": 0.2911568135023117,
"loss_toxic": 0.1034907728433609
},
{
"loss": 2.952083206176758,
"loss_size": 0.6657505482435226,
"loss_pdi": 0.503642150759697,
"loss_ee": 0.7552536249160766,
"loss_delivery": 0.6090254753828048,
"loss_biodist": 0.2907521277666092,
"loss_toxic": 0.14526981711387635
}
],
"val": [
{
"loss": 180.1568603515625,
"loss_size": 30.35355854034424,
"loss_pdi": 10.349318504333496,
"loss_ee": 16.99166774749756,
"loss_delivery": 105.2586441040039,
"loss_biodist": 16.125561237335205,
"loss_toxic": 2.3659912204311695,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.9444444444444444
},
{
"loss": 98.0433578491211,
"loss_size": 12.123669147491455,
"loss_pdi": 7.639462947845459,
"loss_ee": 10.20323395729065,
"loss_delivery": 50.71119499206543,
"loss_biodist": 15.157918453216553,
"loss_toxic": 3.53872871398562,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.9444444444444444
},
{
"loss": 46.07774353027344,
"loss_size": 4.886205315589905,
"loss_pdi": 5.7497618198394775,
"loss_ee": 4.100735902786255,
"loss_delivery": 19.07272720336914,
"loss_biodist": 10.412020444869995,
"loss_toxic": 2.687594591083325,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.9444444444444444
},
{
"loss": 7.691779375076294,
"loss_size": 1.0155234038829803,
"loss_pdi": 1.5577597320079803,
"loss_ee": 1.1349245458841324,
"loss_delivery": 2.3775566816329956,
"loss_biodist": 1.1973169147968292,
"loss_toxic": 0.5206407904624939,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.5151515151515151,
"acc_toxic": 0.9444444444444444
},
{
"loss": 3.2049951553344727,
"loss_size": 0.22598860412836075,
"loss_pdi": 0.3659691661596298,
"loss_ee": 0.5037907660007477,
"loss_delivery": 0.6575061194598675,
"loss_biodist": 0.14315331541001797,
"loss_toxic": 1.2986260950565338,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.2027745246887207,
"loss_size": 0.3320153057575226,
"loss_pdi": 0.31908687576651573,
"loss_ee": 0.48964712768793106,
"loss_delivery": 0.6939206719398499,
"loss_biodist": 0.11092910543084145,
"loss_toxic": 1.241960883140564,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.2450889348983765,
"loss_size": 0.5146260112524033,
"loss_pdi": 0.3309343531727791,
"loss_ee": 0.4984405320137739,
"loss_delivery": 0.698911489918828,
"loss_biodist": 0.12827869690954685,
"loss_toxic": 1.0599181726574898,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.2618945837020874,
"loss_size": 0.49598439037799835,
"loss_pdi": 0.33259066939353943,
"loss_ee": 0.48113041557371616,
"loss_delivery": 0.6671498813666403,
"loss_biodist": 0.11366008035838604,
"loss_toxic": 1.152123749256134,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.264413833618164,
"loss_size": 0.4516768902540207,
"loss_pdi": 0.31526341661810875,
"loss_ee": 0.4847272466868162,
"loss_delivery": 0.6873423364013433,
"loss_biodist": 0.10276161972433329,
"loss_toxic": 1.1997544914484024,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.2070666551589966,
"loss_size": 0.4189356863498688,
"loss_pdi": 0.3158247582614422,
"loss_ee": 0.4858055841177702,
"loss_delivery": 0.7019290700554848,
"loss_biodist": 0.10100412555038929,
"loss_toxic": 1.1597005352377892,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.1986461877822876,
"loss_size": 0.41443589329719543,
"loss_pdi": 0.31525107845664024,
"loss_ee": 0.4934320505708456,
"loss_delivery": 0.7314718030393124,
"loss_biodist": 0.10318759735673666,
"loss_toxic": 1.1180002689361572,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.2213516235351562,
"loss_size": 0.41614747047424316,
"loss_pdi": 0.3155190795660019,
"loss_ee": 0.4953059032559395,
"loss_delivery": 0.7345775403082371,
"loss_biodist": 0.1033031428232789,
"loss_toxic": 1.1332580968737602,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.2576032876968384,
"loss_size": 0.41174091398715973,
"loss_pdi": 0.31428077816963196,
"loss_ee": 0.4866994358599186,
"loss_delivery": 0.7275451384484768,
"loss_biodist": 0.10197119694203138,
"loss_toxic": 1.1903223991394043,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.8888888888888888
}
]
}

1
models/abl/baseline/20260623_205158/outer_fold_4/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [1, 2, 3, 4, 6, 7, 8, 9, 10, 12, 13, 14, 15, 16, 17, 18, 19, 20, 22, 23, 25, 26, 28, 29, 31, 33, 34, 35, 37, 41, 42, 43, 44, 45, 46, 48, 49, 50, 51, 52, 53, 54, 56, 57, 59, 60, 62, 63, 64, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 77, 78, 79, 80, 81, 82, 83, 84, 85, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 101, 102, 103, 104, 105, 107, 108, 110, 111, 112, 113, 114, 116, 117, 119, 120, 121, 122, 123, 124, 126, 127, 128, 130, 132, 133, 134, 135, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 149, 152, 154, 155, 157, 160, 161, 162, 163, 164, 165, 166, 167, 169, 170, 171, 173, 174, 175, 176, 177, 178, 180, 181, 182, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 203, 205, 206, 207, 208, 209, 211, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 229, 230, 231, 232, 234, 235, 236, 238, 239, 240, 241, 243, 244, 245, 246, 247, 248, 249, 250, 251, 253, 254, 257, 258, 260, 261, 262, 263, 264, 265, 266, 267, 270, 271, 272, 273, 277, 278, 279, 280, 281, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 298, 299, 300, 301, 302, 303, 304, 305, 306, 309, 311, 312, 313, 314, 315, 316, 318, 319, 320, 322, 323, 324, 325, 326, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 341, 342, 343, 345, 346, 347, 349, 350, 351, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 365, 367, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 381, 383, 384, 385, 386, 388, 390, 391, 392, 393, 394, 395, 396, 397, 399, 400, 401, 402, 403, 405, 406, 407, 408, 409, 410, 411, 413, 415, 416, 417, 418, 419], "outer_test_idx": [0, 5, 11, 21, 24, 27, 30, 32, 36, 38, 39, 40, 47, 55, 58, 61, 65, 76, 86, 95, 100, 106, 109, 115, 118, 125, 129, 131, 136, 148, 150, 151, 153, 156, 158, 159, 168, 172, 179, 183, 184, 185, 202, 204, 210, 212, 213, 228, 233, 237, 242, 252, 255, 256, 259, 268, 269, 274, 275, 276, 282, 297, 307, 308, 310, 317, 321, 327, 340, 344, 348, 352, 364, 366, 368, 379, 380, 382, 387, 389, 398, 404, 412, 414]}

42
models/abl/baseline/20260623_205158/outer_fold_4/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.6868306794055117,
"rmse": 0.8287524838005083,
"mae": 0.5001247360376597,
"r2": 0.08093523565734784
},
"delivery": {
"n_samples": 58,
"mse": 1.1008487633192388,
"rmse": 1.0492134021824344,
"mae": 0.654239107951008,
"r2": -0.06450258120608798
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6752941176470588,
"recall": 0.718475073313783,
"f1": 0.6785714285714286
},
"ee": {
"n_samples": 84,
"accuracy": 0.6785714285714286,
"precision": 0.626984126984127,
"recall": 0.6594017094017094,
"f1": 0.6342062193126022
},
"toxic": {
"n_samples": 59,
"accuracy": 0.9661016949152542,
"precision": 0.8,
"recall": 0.9821428571428572,
"f1": 0.865909090909091
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.2992118497501824,
"js_divergence": 0.08211837700207104
}
}

12
models/abl/baseline/20260623_205158/repeat_1/outer_fold_0/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.18569939775458322,
"lr": 0.0009872295492725695,
"weight_decay": 0.017393516831905452,
"backbone_lr_ratio": 0.17239541174804215,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_0/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 11}

237
models/abl/baseline/20260623_205158/repeat_1/outer_fold_0/history.json Normal file
View File

@ -0,0 +1,237 @@
{
"train": [
{
"loss": 5.394427347183227,
"loss_size": 0.8760069787502289,
"loss_pdi": 0.6829078137874603,
"loss_ee": 1.0850867986679078,
"loss_delivery": 1.0246612578630447,
"loss_biodist": 1.1033188700675964,
"loss_toxic": 0.630187314748764
},
{
"loss": 4.4071554899215695,
"loss_size": 0.7775529831647873,
"loss_pdi": 0.6168016731739044,
"loss_ee": 0.9830431044101715,
"loss_delivery": 0.9110523700714112,
"loss_biodist": 0.7133126735687256,
"loss_toxic": 0.42119770050048827
},
{
"loss": 3.8539650201797486,
"loss_size": 0.7501171737909317,
"loss_pdi": 0.585977566242218,
"loss_ee": 0.9200241446495057,
"loss_delivery": 0.7962285697460174,
"loss_biodist": 0.5048318803310394,
"loss_toxic": 0.3135771378874779
},
{
"loss": 3.6865764617919923,
"loss_size": 0.7014920532703399,
"loss_pdi": 0.5886935204267502,
"loss_ee": 0.9026208221912384,
"loss_delivery": 0.8553118914365768,
"loss_biodist": 0.4308844119310379,
"loss_toxic": 0.22991862669587135
},
{
"loss": 3.268519568443298,
"loss_size": 0.658152274787426,
"loss_pdi": 0.5069636017084121,
"loss_ee": 0.830296790599823,
"loss_delivery": 0.7789995819330215,
"loss_biodist": 0.36636197566986084,
"loss_toxic": 0.14613031409680843
},
{
"loss": 2.986606740951538,
"loss_size": 0.5903288200497627,
"loss_pdi": 0.5061607718467712,
"loss_ee": 0.7544377148151398,
"loss_delivery": 0.7368057832121849,
"loss_biodist": 0.30559518486261367,
"loss_toxic": 0.10927746798843145
},
{
"loss": 2.9839526653289794,
"loss_size": 0.5620119944214821,
"loss_pdi": 0.478989964723587,
"loss_ee": 0.7554892122745513,
"loss_delivery": 0.8041216015815735,
"loss_biodist": 0.2546705946326256,
"loss_toxic": 0.1464777808636427
},
{
"loss": 2.7193713426589965,
"loss_size": 0.5286610335111618,
"loss_pdi": 0.45270849764347076,
"loss_ee": 0.7556517481803894,
"loss_delivery": 0.68291757106781,
"loss_biodist": 0.2408062547445297,
"loss_toxic": 0.07487203162163496
},
{
"loss": 2.623748242855072,
"loss_size": 0.5352809801697731,
"loss_pdi": 0.45280796885490415,
"loss_ee": 0.7069973200559616,
"loss_delivery": 0.638873553276062,
"loss_biodist": 0.22641604393720627,
"loss_toxic": 0.07768626157194376
},
{
"loss": 2.581704306602478,
"loss_size": 0.4847213253378868,
"loss_pdi": 0.4549128830432892,
"loss_ee": 0.717718631029129,
"loss_delivery": 0.6322640471160412,
"loss_biodist": 0.21297531425952912,
"loss_toxic": 0.09345897333696485
},
{
"loss": 2.552983558177948,
"loss_size": 0.4678229957818985,
"loss_pdi": 0.45056481957435607,
"loss_ee": 0.7299788951873779,
"loss_delivery": 0.6316331580281258,
"loss_biodist": 0.21403777301311494,
"loss_toxic": 0.07426234278827906
}
],
"val": [
{
"loss": 310.3435287475586,
"loss_size": 222.49696350097656,
"loss_pdi": 1.003352914005518,
"loss_ee": 3.3513469035897288,
"loss_delivery": 78.34090423583984,
"loss_biodist": 6.392218112945557,
"loss_toxic": 0.6385428309440613,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.21212121212121213,
"acc_toxic": 0.8846153846153846
},
{
"loss": 141.68408203125,
"loss_size": 114.88703918457031,
"loss_pdi": 2.3383941004940425,
"loss_ee": 2.82101229543332,
"loss_delivery": 14.252228736877441,
"loss_biodist": 6.809396266937256,
"loss_toxic": 1.9328919649124146,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.21212121212121213,
"acc_toxic": 0.8846153846153846
},
{
"loss": 40.947078704833984,
"loss_size": 25.30094623565674,
"loss_pdi": 2.101527013015584,
"loss_ee": 1.5821967422962189,
"loss_delivery": 6.3904619216918945,
"loss_biodist": 4.2740678787231445,
"loss_toxic": 1.8244426250457764,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.21212121212121213,
"acc_toxic": 0.8846153846153846
},
{
"loss": 6.22724986076355,
"loss_size": 1.4807607233524323,
"loss_pdi": 1.0755788842216134,
"loss_ee": 0.8111673966050148,
"loss_delivery": 1.914881706237793,
"loss_biodist": 0.7986646890640259,
"loss_toxic": 0.23427508771419525,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.9230769230769231
},
{
"loss": 2.032481074333191,
"loss_size": 0.18289357819594443,
"loss_pdi": 0.4539054110646248,
"loss_ee": 0.6869342923164368,
"loss_delivery": 0.5226569175720215,
"loss_biodist": 0.12324852496385574,
"loss_toxic": 0.06994590163230896,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9615384615384616
},
{
"loss": 1.9082787334918976,
"loss_size": 0.17314767945208587,
"loss_pdi": 0.44033991545438766,
"loss_ee": 0.6851533055305481,
"loss_delivery": 0.44304659962654114,
"loss_biodist": 0.1133018285036087,
"loss_toxic": 0.05914802849292755,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 0.9615384615384616
},
{
"loss": 1.9223107695579529,
"loss_size": 0.16595131054054946,
"loss_pdi": 0.42985038459300995,
"loss_ee": 0.6594338417053223,
"loss_delivery": 0.41970717906951904,
"loss_biodist": 0.10797291994094849,
"loss_toxic": 0.14298951625823975,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9615384615384616
},
{
"loss": 1.8832128643989563,
"loss_size": 0.1460553091019392,
"loss_pdi": 0.42990442365407944,
"loss_ee": 0.6492694467306137,
"loss_delivery": 0.41635453701019287,
"loss_biodist": 0.11513877660036087,
"loss_toxic": 0.13046549260616302,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9615384615384616
},
{
"loss": 1.879135638475418,
"loss_size": 0.14139589853584766,
"loss_pdi": 0.44527293741703033,
"loss_ee": 0.6727772951126099,
"loss_delivery": 0.415428102016449,
"loss_biodist": 0.11692274361848831,
"loss_toxic": 0.09393423795700073,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9615384615384616
},
{
"loss": 1.8444421291351318,
"loss_size": 0.13496902398765087,
"loss_pdi": 0.4370863810181618,
"loss_ee": 0.6553614735603333,
"loss_delivery": 0.4066224992275238,
"loss_biodist": 0.11289884895086288,
"loss_toxic": 0.10283103585243225,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 0.9615384615384616
},
{
"loss": 1.8573571145534515,
"loss_size": 0.1297498056665063,
"loss_pdi": 0.4353114068508148,
"loss_ee": 0.6571271121501923,
"loss_delivery": 0.4090708792209625,
"loss_biodist": 0.11798186600208282,
"loss_toxic": 0.11338865011930466,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 0.9615384615384616
}
]
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_0/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 21, 22, 23, 24, 25, 26, 27, 28, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 52, 53, 54, 55, 56, 57, 58, 61, 62, 63, 64, 65, 66, 67, 68, 71, 72, 73, 75, 76, 77, 80, 81, 85, 86, 88, 89, 90, 92, 94, 95, 96, 98, 99, 100, 101, 103, 104, 105, 106, 107, 108, 109, 111, 113, 114, 115, 116, 117, 118, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 133, 134, 135, 136, 138, 139, 141, 144, 145, 146, 147, 148, 149, 150, 151, 153, 154, 155, 156, 157, 158, 159, 160, 163, 164, 167, 168, 169, 170, 171, 172, 175, 176, 178, 179, 180, 182, 183, 184, 185, 188, 189, 190, 191, 192, 193, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 206, 208, 210, 211, 212, 213, 216, 217, 218, 219, 220, 222, 224, 225, 227, 228, 229, 230, 231, 232, 233, 234, 236, 237, 239, 241, 242, 244, 245, 247, 248, 249, 250, 252, 253, 255, 256, 257, 258, 259, 260, 261, 262, 265, 267, 268, 269, 270, 271, 272, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 285, 286, 287, 288, 291, 292, 294, 296, 297, 298, 299, 300, 301, 303, 304, 306, 307, 308, 309, 310, 311, 312, 313, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 332, 333, 335, 336, 337, 338, 339, 340, 342, 343, 344, 345, 346, 347, 348, 349, 351, 352, 354, 355, 356, 357, 358, 359, 360, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 375, 377, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 393, 395, 396, 397, 398, 399, 400, 401, 403, 404, 405, 406, 407, 408, 409, 411, 412, 413, 414, 415, 417, 418, 419], "outer_test_idx": [7, 16, 20, 29, 51, 59, 60, 69, 70, 74, 78, 79, 82, 83, 84, 87, 91, 93, 97, 102, 110, 112, 119, 120, 121, 132, 137, 140, 142, 143, 152, 161, 162, 165, 166, 173, 174, 177, 181, 186, 187, 194, 205, 207, 209, 214, 215, 221, 223, 226, 235, 238, 240, 243, 246, 251, 254, 263, 264, 266, 273, 284, 289, 290, 293, 295, 302, 305, 314, 315, 331, 334, 341, 350, 353, 361, 373, 376, 378, 392, 394, 402, 410, 416]}

42
models/abl/baseline/20260623_205158/repeat_1/outer_fold_0/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 83,
"mse": 1.5000130580754827,
"rmse": 1.2247502023169796,
"mae": 0.5513235701315374,
"r2": 0.08195508578319866
},
"delivery": {
"n_samples": 58,
"mse": 0.6098699437647568,
"rmse": 0.7809417031794094,
"mae": 0.6011813956088033,
"r2": 0.2248184995335506
},
"pdi": {
"n_samples": 84,
"accuracy": 0.8333333333333334,
"precision": 0.7904490377761939,
"recall": 0.7904490377761939,
"f1": 0.7904490377761939
},
"ee": {
"n_samples": 84,
"accuracy": 0.6904761904761905,
"precision": 0.6475648393751842,
"recall": 0.6809523809523809,
"f1": 0.6526484751203853
},
"toxic": {
"n_samples": 58,
"accuracy": 0.9827586206896551,
"precision": 0.9912280701754386,
"recall": 0.75,
"f1": 0.8289085545722714
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.31929652154466054,
"js_divergence": 0.06311757696370286
}
}

12
models/abl/baseline/20260623_205158/repeat_1/outer_fold_1/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.36640960296080094,
"lr": 0.0009935032389272061,
"weight_decay": 0.007506834667713186,
"backbone_lr_ratio": 0.10108281143505408,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_1/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 11}

237
models/abl/baseline/20260623_205158/repeat_1/outer_fold_1/history.json Normal file
View File

@ -0,0 +1,237 @@
{
"train": [
{
"loss": 5.690161800384521,
"loss_size": 1.189884325861931,
"loss_pdi": 0.6715496301651,
"loss_ee": 1.0822121977806092,
"loss_delivery": 0.9816046833992005,
"loss_biodist": 1.1128089606761933,
"loss_toxic": 0.6609058678150177
},
{
"loss": 5.015087723731995,
"loss_size": 1.1365506678819657,
"loss_pdi": 0.6081841349601745,
"loss_ee": 0.9931918501853942,
"loss_delivery": 0.9461931824684143,
"loss_biodist": 0.7874724328517914,
"loss_toxic": 0.5648481160402298
},
{
"loss": 4.4928162574768065,
"loss_size": 1.0737243711948394,
"loss_pdi": 0.5840542733669281,
"loss_ee": 0.9511588513851166,
"loss_delivery": 0.8004588901996612,
"loss_biodist": 0.619153293967247,
"loss_toxic": 0.48818554282188414
},
{
"loss": 4.119502520561218,
"loss_size": 1.052136318385601,
"loss_pdi": 0.5726546883583069,
"loss_ee": 0.9047514855861664,
"loss_delivery": 0.773554441332817,
"loss_biodist": 0.4987696439027786,
"loss_toxic": 0.3413187012076378
},
{
"loss": 3.8408268690109253,
"loss_size": 1.0080846071243286,
"loss_pdi": 0.593186417222023,
"loss_ee": 0.8830170869827271,
"loss_delivery": 0.7021539807319641,
"loss_biodist": 0.4199835598468781,
"loss_toxic": 0.25591021180152895
},
{
"loss": 3.603969860076904,
"loss_size": 0.9665896967053413,
"loss_pdi": 0.534265223145485,
"loss_ee": 0.8248504996299744,
"loss_delivery": 0.6819487854838371,
"loss_biodist": 0.3866705894470215,
"loss_toxic": 0.22798593267798423
},
{
"loss": 3.5279754161834718,
"loss_size": 0.9392264425754547,
"loss_pdi": 0.5317829042673111,
"loss_ee": 0.798810213804245,
"loss_delivery": 0.747761806845665,
"loss_biodist": 0.34822759628295896,
"loss_toxic": 0.18188186809420587
},
{
"loss": 3.661241555213928,
"loss_size": 1.114355704188347,
"loss_pdi": 0.544609448313713,
"loss_ee": 0.7809613883495331,
"loss_delivery": 0.6891857922077179,
"loss_biodist": 0.35454697012901304,
"loss_toxic": 0.2011748395860195
},
{
"loss": 3.647503685951233,
"loss_size": 1.1867847949266435,
"loss_pdi": 0.49068042635917664,
"loss_ee": 0.7612938404083252,
"loss_delivery": 0.7279756560921669,
"loss_biodist": 0.3450884252786636,
"loss_toxic": 0.16020391061902045
},
{
"loss": 3.4281585454940795,
"loss_size": 0.9419048264622688,
"loss_pdi": 0.498330682516098,
"loss_ee": 0.7777611792087555,
"loss_delivery": 0.7194288223981857,
"loss_biodist": 0.33402347564697266,
"loss_toxic": 0.1774237260222435
},
{
"loss": 3.402118945121765,
"loss_size": 0.9237892001867294,
"loss_pdi": 0.4987943768501282,
"loss_ee": 0.7891723573207855,
"loss_delivery": 0.7009412780404091,
"loss_biodist": 0.3317346051335335,
"loss_toxic": 0.17829641103744506
}
],
"val": [
{
"loss": 14.251262001693249,
"loss_size": 2.7426084615290165,
"loss_pdi": 2.7668208479703935,
"loss_ee": 1.84965006634593,
"loss_delivery": 0.7406838536262512,
"loss_biodist": 3.6853599548339844,
"loss_toxic": 2.5724637508392334,
"acc_pdi": 0.30303030303030304,
"acc_ee": 0.15151515151515152,
"acc_toxic": 0.9545454545454546
},
{
"loss": 13.879438638687134,
"loss_size": 3.246104061603546,
"loss_pdi": 1.1581786987371743,
"loss_ee": 1.8079487681388855,
"loss_delivery": 0.34908974170684814,
"loss_biodist": 4.72705078125,
"loss_toxic": 2.7753286361694336,
"acc_pdi": 0.30303030303030304,
"acc_ee": 0.24242424242424243,
"acc_toxic": 0.9545454545454546
},
{
"loss": 11.30401086807251,
"loss_size": 3.140974283218384,
"loss_pdi": 0.590329721570015,
"loss_ee": 2.473829984664917,
"loss_delivery": 0.9451518058776855,
"loss_biodist": 2.8875391483306885,
"loss_toxic": 1.450668454170227,
"acc_pdi": 0.30303030303030304,
"acc_ee": 0.24242424242424243,
"acc_toxic": 0.9545454545454546
},
{
"loss": 5.917563438415527,
"loss_size": 2.3373581171035767,
"loss_pdi": 0.7212768197059631,
"loss_ee": 1.3266438841819763,
"loss_delivery": 0.5801224112510681,
"loss_biodist": 0.8109126687049866,
"loss_toxic": 0.22340703010559082,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.36363636363636365,
"acc_toxic": 0.9545454545454546
},
{
"loss": 2.2017491459846497,
"loss_size": 0.4596553444862366,
"loss_pdi": 0.5928162932395935,
"loss_ee": 0.5837425142526627,
"loss_delivery": 0.2711705267429352,
"loss_biodist": 0.2295028567314148,
"loss_toxic": 0.06653276830911636,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 1.0
},
{
"loss": 1.843619018793106,
"loss_size": 0.337517112493515,
"loss_pdi": 0.5240353643894196,
"loss_ee": 0.5068006366491318,
"loss_delivery": 0.22263692319393158,
"loss_biodist": 0.20966367423534393,
"loss_toxic": 0.03639179468154907,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 1.8369120955467224,
"loss_size": 0.40816980600357056,
"loss_pdi": 0.5170520693063736,
"loss_ee": 0.49596188589930534,
"loss_delivery": 0.19062940776348114,
"loss_biodist": 0.18511846661567688,
"loss_toxic": 0.03358370065689087,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 1.6854368150234222,
"loss_size": 0.2946223318576813,
"loss_pdi": 0.5051011890172958,
"loss_ee": 0.4803689830005169,
"loss_delivery": 0.18430842459201813,
"loss_biodist": 0.18295420706272125,
"loss_toxic": 0.02873709797859192,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 1.6858542263507843,
"loss_size": 0.305390328168869,
"loss_pdi": 0.5007014721632004,
"loss_ee": 0.47501955181360245,
"loss_delivery": 0.18350207805633545,
"loss_biodist": 0.18312783539295197,
"loss_toxic": 0.028951579704880714,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 1.6575334668159485,
"loss_size": 0.2784598469734192,
"loss_pdi": 0.5052449852228165,
"loss_ee": 0.4691324532032013,
"loss_delivery": 0.18377985060214996,
"loss_biodist": 0.1851455122232437,
"loss_toxic": 0.026173150166869164,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 1.647377073764801,
"loss_size": 0.25929317623376846,
"loss_pdi": 0.5161628872156143,
"loss_ee": 0.4699825271964073,
"loss_delivery": 0.18454687297344208,
"loss_biodist": 0.18249762058258057,
"loss_toxic": 0.025294104591012,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
}
]
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_1/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 3, 5, 6, 7, 10, 11, 12, 13, 15, 16, 18, 19, 20, 21, 24, 26, 27, 29, 30, 31, 32, 33, 35, 36, 38, 39, 40, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 65, 66, 67, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 97, 98, 99, 100, 101, 102, 103, 105, 106, 107, 108, 109, 110, 112, 114, 115, 118, 119, 120, 121, 123, 124, 125, 126, 128, 129, 130, 131, 132, 136, 137, 140, 141, 142, 143, 144, 145, 146, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 161, 162, 164, 165, 166, 167, 168, 169, 171, 172, 173, 174, 175, 176, 177, 178, 179, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 194, 195, 196, 197, 198, 199, 201, 202, 204, 205, 206, 207, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 225, 226, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 240, 242, 243, 244, 246, 247, 248, 250, 251, 252, 254, 255, 256, 258, 259, 260, 261, 262, 263, 264, 266, 268, 269, 270, 271, 273, 274, 275, 276, 277, 278, 279, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 295, 296, 297, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 313, 314, 315, 316, 317, 318, 319, 321, 322, 323, 324, 325, 327, 328, 329, 330, 331, 333, 334, 335, 337, 338, 339, 340, 341, 342, 343, 344, 345, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 361, 362, 364, 365, 366, 367, 368, 369, 373, 374, 375, 376, 378, 379, 380, 381, 382, 384, 387, 388, 389, 391, 392, 393, 394, 396, 397, 398, 399, 402, 403, 404, 405, 406, 408, 409, 410, 411, 412, 413, 414, 416, 417, 419], "outer_test_idx": [1, 2, 4, 8, 9, 14, 17, 22, 23, 25, 28, 34, 37, 41, 43, 45, 63, 64, 68, 85, 96, 104, 111, 113, 116, 117, 122, 127, 133, 134, 135, 138, 139, 147, 160, 163, 170, 180, 191, 192, 193, 200, 203, 208, 224, 227, 230, 239, 241, 245, 249, 253, 257, 265, 267, 272, 280, 294, 298, 299, 309, 312, 320, 326, 332, 336, 346, 347, 360, 363, 370, 371, 372, 377, 383, 385, 386, 390, 395, 400, 401, 407, 415, 418]}

42
models/abl/baseline/20260623_205158/repeat_1/outer_fold_1/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.35953046688813006,
"rmse": 0.5996085947417116,
"mae": 0.4586696324404329,
"r2": -0.09787822464275231
},
"delivery": {
"n_samples": 61,
"mse": 1.1161931438403756,
"rmse": 1.0565004230194968,
"mae": 0.6922882909840736,
"r2": 0.15512225528421297
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7261904761904762,
"precision": 0.6685185185185185,
"recall": 0.7063492063492063,
"f1": 0.6762192056309704
},
"ee": {
"n_samples": 84,
"accuracy": 0.5476190476190477,
"precision": 0.48175125313283207,
"recall": 0.48011512297226583,
"f1": 0.4721157939548744
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9508196721311475,
"precision": 0.75,
"recall": 0.9741379310344828,
"f1": 0.8200589970501475
},
"biodist": {
"n_samples": 61,
"kl_divergence": 0.3950644341107201,
"js_divergence": 0.09966091530608283
}
}

12
models/abl/baseline/20260623_205158/repeat_1/outer_fold_2/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.318342405407684,
"lr": 0.0008937362022799393,
"weight_decay": 0.06633867694372529,
"backbone_lr_ratio": 0.20906844321970927,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_2/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 10}

216
models/abl/baseline/20260623_205158/repeat_1/outer_fold_2/history.json Normal file
View File

@ -0,0 +1,216 @@
{
"train": [
{
"loss": 6.120371389389038,
"loss_size": 1.5454729944467545,
"loss_pdi": 0.6601403653621674,
"loss_ee": 1.069458258152008,
"loss_delivery": 1.0308290749788285,
"loss_biodist": 1.1528971076011658,
"loss_toxic": 0.6679352998733521
},
{
"loss": 4.989258742332458,
"loss_size": 1.1054174512624741,
"loss_pdi": 0.6432794153690338,
"loss_ee": 1.013900339603424,
"loss_delivery": 0.890376278758049,
"loss_biodist": 0.8432588100433349,
"loss_toxic": 0.5088101089000702
},
{
"loss": 4.480851435661316,
"loss_size": 1.0250093176960946,
"loss_pdi": 0.6028325498104096,
"loss_ee": 0.9736166954040527,
"loss_delivery": 0.8690223038196564,
"loss_biodist": 0.6110427618026734,
"loss_toxic": 0.4193166673183441
},
{
"loss": 4.458040189743042,
"loss_size": 1.3500118359923363,
"loss_pdi": 0.5795828342437744,
"loss_ee": 0.9031877458095551,
"loss_delivery": 0.8723745167255401,
"loss_biodist": 0.5005194991827011,
"loss_toxic": 0.2762876100838184
},
{
"loss": 3.6713650465011596,
"loss_size": 0.9428067192435264,
"loss_pdi": 0.5491054594516754,
"loss_ee": 0.8509662389755249,
"loss_delivery": 0.7096938818693161,
"loss_biodist": 0.42276733219623563,
"loss_toxic": 0.21214713156223297
},
{
"loss": 3.55607213973999,
"loss_size": 0.9256311297416687,
"loss_pdi": 0.5255982667207718,
"loss_ee": 0.8466547906398774,
"loss_delivery": 0.7089663371443748,
"loss_biodist": 0.41232300698757174,
"loss_toxic": 0.15456384383141994
},
{
"loss": 3.4522277116775513,
"loss_size": 0.8846466436982154,
"loss_pdi": 0.5214570879936218,
"loss_ee": 0.7946006238460541,
"loss_delivery": 0.752051842212677,
"loss_biodist": 0.38330944180488585,
"loss_toxic": 0.13409302458167077
},
{
"loss": 3.309400725364685,
"loss_size": 0.8188066214323044,
"loss_pdi": 0.5153862059116363,
"loss_ee": 0.8008714616298676,
"loss_delivery": 0.6458104610443115,
"loss_biodist": 0.4099011570215225,
"loss_toxic": 0.13527849614620208
},
{
"loss": 3.2613839387893675,
"loss_size": 0.843678742647171,
"loss_pdi": 0.5112991333007812,
"loss_ee": 0.7749116897583008,
"loss_delivery": 0.6798483833670617,
"loss_biodist": 0.35379023253917696,
"loss_toxic": 0.11434709765017033
},
{
"loss": 3.2790340423583983,
"loss_size": 0.8493687808513641,
"loss_pdi": 0.5153828114271164,
"loss_ee": 0.7604458868503571,
"loss_delivery": 0.7082577548921108,
"loss_biodist": 0.3487837702035904,
"loss_toxic": 0.11392149552702904
}
],
"val": [
{
"loss": 31.121932983398438,
"loss_size": 11.37618112564087,
"loss_pdi": 7.627802848815918,
"loss_ee": 8.261778354644775,
"loss_delivery": 0.7883476614952087,
"loss_biodist": 3.215151071548462,
"loss_toxic": 6.26379405730404e-05,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 27.223604202270508,
"loss_size": 8.448630094528198,
"loss_pdi": 4.6595776081085205,
"loss_ee": 7.333366632461548,
"loss_delivery": 2.713754892349243,
"loss_biodist": 4.367090225219727,
"loss_toxic": 2.7362952096154913e-05,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 13.467486381530762,
"loss_size": 1.722923556342721,
"loss_pdi": 2.8058156967163086,
"loss_ee": 4.767172336578369,
"loss_delivery": 1.4759490489959717,
"loss_biodist": 2.927734375,
"loss_toxic": 0.00035144996945746243,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.21212121212121213,
"acc_toxic": 1.0
},
{
"loss": 5.715693235397339,
"loss_size": 0.7490728795528412,
"loss_pdi": 1.3319993615150452,
"loss_ee": 1.9729333519935608,
"loss_delivery": 0.9985048174858093,
"loss_biodist": 0.6482511162757874,
"loss_toxic": 0.10335085541009903,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.3333333333333333,
"acc_toxic": 0.9130434782608695
},
{
"loss": 2.4365017414093018,
"loss_size": 0.3043541982769966,
"loss_pdi": 0.45934683084487915,
"loss_ee": 0.4923722743988037,
"loss_delivery": 0.7170434594154358,
"loss_biodist": 0.21668121218681335,
"loss_toxic": 0.25241619348526,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.7272727272727273,
"acc_toxic": 0.9130434782608695
},
{
"loss": 2.2742066383361816,
"loss_size": 0.22480389149859548,
"loss_pdi": 0.39823178201913834,
"loss_ee": 0.4579532891511917,
"loss_delivery": 0.7410005331039429,
"loss_biodist": 0.24491435289382935,
"loss_toxic": 0.2105923891067505,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9130434782608695
},
{
"loss": 2.2190119475126266,
"loss_size": 0.2139543890953064,
"loss_pdi": 0.38967230170965195,
"loss_ee": 0.4544268883764744,
"loss_delivery": 0.729695200920105,
"loss_biodist": 0.24956533312797546,
"loss_toxic": 0.1845780611038208,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9130434782608695
},
{
"loss": 2.189032658934593,
"loss_size": 0.18267557537183166,
"loss_pdi": 0.3697962909936905,
"loss_ee": 0.4524207003414631,
"loss_delivery": 0.7574172019958496,
"loss_biodist": 0.26046401262283325,
"loss_toxic": 0.16923433542251587,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9130434782608695
},
{
"loss": 2.1914793998003006,
"loss_size": 0.18211279627939803,
"loss_pdi": 0.3656071238219738,
"loss_ee": 0.45181378722190857,
"loss_delivery": 0.7655497193336487,
"loss_biodist": 0.2546330690383911,
"loss_toxic": 0.17467162013053894,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9130434782608695
},
{
"loss": 2.1843226104974747,
"loss_size": 0.1741333045065403,
"loss_pdi": 0.3657117150723934,
"loss_ee": 0.4539703540503979,
"loss_delivery": 0.7712145447731018,
"loss_biodist": 0.2501569390296936,
"loss_toxic": 0.17209932208061218,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9130434782608695
}
]
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_2/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 13, 14, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 32, 33, 34, 36, 37, 38, 39, 40, 41, 43, 44, 45, 47, 48, 49, 50, 51, 55, 56, 57, 58, 59, 60, 61, 63, 64, 65, 68, 69, 70, 72, 74, 76, 78, 79, 80, 82, 83, 84, 85, 86, 87, 88, 91, 93, 94, 95, 96, 97, 100, 101, 102, 103, 104, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 142, 143, 145, 147, 148, 149, 150, 151, 152, 153, 156, 157, 158, 159, 160, 161, 162, 163, 165, 166, 167, 168, 170, 171, 172, 173, 174, 175, 177, 179, 180, 181, 183, 184, 185, 186, 187, 188, 191, 192, 193, 194, 196, 197, 199, 200, 202, 203, 204, 205, 207, 208, 209, 210, 211, 212, 213, 214, 215, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 232, 233, 235, 236, 237, 238, 239, 240, 241, 242, 243, 245, 246, 249, 250, 251, 252, 253, 254, 255, 256, 257, 259, 260, 261, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 280, 281, 282, 283, 284, 285, 287, 289, 290, 293, 294, 295, 296, 297, 298, 299, 300, 302, 303, 305, 307, 308, 309, 310, 311, 312, 313, 314, 315, 317, 318, 320, 321, 322, 323, 324, 326, 327, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 358, 360, 361, 363, 364, 365, 366, 368, 369, 370, 371, 372, 373, 376, 377, 378, 379, 380, 381, 382, 383, 385, 386, 387, 388, 389, 390, 392, 394, 395, 398, 399, 400, 401, 402, 404, 405, 407, 408, 409, 410, 412, 413, 414, 415, 416, 417, 418], "outer_test_idx": [10, 12, 15, 31, 35, 42, 46, 52, 53, 54, 62, 66, 67, 71, 73, 75, 77, 81, 89, 90, 92, 98, 99, 105, 123, 124, 141, 144, 146, 154, 155, 164, 169, 176, 178, 182, 189, 190, 195, 198, 201, 206, 216, 217, 218, 219, 231, 234, 244, 247, 248, 258, 262, 270, 277, 278, 279, 286, 288, 291, 292, 301, 304, 306, 316, 319, 325, 328, 343, 357, 359, 362, 367, 374, 375, 384, 391, 393, 396, 397, 403, 406, 411, 419]}

42
models/abl/baseline/20260623_205158/repeat_1/outer_fold_2/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.4926918006241934,
"rmse": 0.7019200813655307,
"mae": 0.537246183603115,
"r2": 0.026942496675384175
},
"delivery": {
"n_samples": 60,
"mse": 0.726897839382552,
"rmse": 0.8525830395818064,
"mae": 0.5536939328614001,
"r2": 0.03237298701263025
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7261904761904762,
"precision": 0.6686520376175549,
"recall": 0.6972140762463344,
"f1": 0.6762192056309704
},
"ee": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.6139316239316239,
"recall": 0.6785714285714285,
"f1": 0.6277576489533011
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9344262295081968,
"precision": 0.7142857142857143,
"recall": 0.9655172413793103,
"f1": 0.7821428571428571
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.38696828264806826,
"js_divergence": 0.0987549780391864
}
}

12
models/abl/baseline/20260623_205158/repeat_1/outer_fold_3/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.12612237693012526,
"lr": 0.0009464809649969671,
"weight_decay": 0.07051438027639334,
"backbone_lr_ratio": 0.9343703462702573,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_3/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 18}

384
models/abl/baseline/20260623_205158/repeat_1/outer_fold_3/history.json Normal file
View File

@ -0,0 +1,384 @@
{
"train": [
{
"loss": 5.2266968250274655,
"loss_size": 0.8537909328937531,
"loss_pdi": 0.6772095084190368,
"loss_ee": 1.075910472869873,
"loss_delivery": 0.9067249834537506,
"loss_biodist": 1.1115365147590637,
"loss_toxic": 0.6086407035589219
},
{
"loss": 4.306142902374267,
"loss_size": 0.7718606650829315,
"loss_pdi": 0.6121319711208344,
"loss_ee": 0.992641133069992,
"loss_delivery": 0.8013194978237153,
"loss_biodist": 0.7360473692417144,
"loss_toxic": 0.4062326669692993
},
{
"loss": 3.8139957427978515,
"loss_size": 0.6974141225218773,
"loss_pdi": 0.5643113672733306,
"loss_ee": 0.9088626861572265,
"loss_delivery": 0.8008200973272324,
"loss_biodist": 0.5445760816335679,
"loss_toxic": 0.3136837497353554
},
{
"loss": 4.050071048736572,
"loss_size": 0.7502574235200882,
"loss_pdi": 0.6060497105121613,
"loss_ee": 0.954151850938797,
"loss_delivery": 0.9302307486534118,
"loss_biodist": 0.5428307026624679,
"loss_toxic": 0.2941423043608665
},
{
"loss": 3.5870924472808836,
"loss_size": 0.7568428218364716,
"loss_pdi": 0.5567073076963425,
"loss_ee": 0.8350918650627136,
"loss_delivery": 0.7545906066894531,
"loss_biodist": 0.5207897424697876,
"loss_toxic": 0.18692626059055328
},
{
"loss": 3.4063260555267334,
"loss_size": 0.8311702013015747,
"loss_pdi": 0.4987951755523682,
"loss_ee": 0.810703581571579,
"loss_delivery": 0.7564232409000397,
"loss_biodist": 0.35182228237390517,
"loss_toxic": 0.1801762593910098
},
{
"loss": 3.16544189453125,
"loss_size": 0.7945439279079437,
"loss_pdi": 0.5021692991256714,
"loss_ee": 0.7565134823322296,
"loss_delivery": 0.6438627123832703,
"loss_biodist": 0.3402824863791466,
"loss_toxic": 0.14987281998619437
},
{
"loss": 2.81198890209198,
"loss_size": 0.6519294038414956,
"loss_pdi": 0.5009945303201675,
"loss_ee": 0.7232840657234192,
"loss_delivery": 0.6078858524560928,
"loss_biodist": 0.23479606062173844,
"loss_toxic": 0.11334207095205784
},
{
"loss": 2.5400012135505676,
"loss_size": 0.5920610263943672,
"loss_pdi": 0.4433219224214554,
"loss_ee": 0.6640033602714539,
"loss_delivery": 0.5541495159268379,
"loss_biodist": 0.21075763255357743,
"loss_toxic": 0.09307465609163046
},
{
"loss": 2.5925466775894166,
"loss_size": 0.609738776087761,
"loss_pdi": 0.4520172834396362,
"loss_ee": 0.6599611103534698,
"loss_delivery": 0.5127524852752685,
"loss_biodist": 0.20787514448165895,
"loss_toxic": 0.16722667180001735
},
{
"loss": 2.3221617102622987,
"loss_size": 0.5035484060645103,
"loss_pdi": 0.43082907795906067,
"loss_ee": 0.6492843389511108,
"loss_delivery": 0.48944447487592696,
"loss_biodist": 0.16787422224879264,
"loss_toxic": 0.10015662806108594
},
{
"loss": 2.348598837852478,
"loss_size": 0.5328022614121437,
"loss_pdi": 0.4518605023622513,
"loss_ee": 0.604455542564392,
"loss_delivery": 0.4886851474642754,
"loss_biodist": 0.18939172700047494,
"loss_toxic": 0.10189135912805795
},
{
"loss": 2.5277572751045225,
"loss_size": 0.8685659632086754,
"loss_pdi": 0.4186414390802383,
"loss_ee": 0.6073152393102645,
"loss_delivery": 0.4253119841217995,
"loss_biodist": 0.14511534199118614,
"loss_toxic": 0.08848735988140106
},
{
"loss": 2.149649214744568,
"loss_size": 0.4845284879207611,
"loss_pdi": 0.40528068989515303,
"loss_ee": 0.5713456183671951,
"loss_delivery": 0.447555610537529,
"loss_biodist": 0.14241116344928742,
"loss_toxic": 0.11715929917991161
},
{
"loss": 2.0795523881912232,
"loss_size": 0.5185737796127796,
"loss_pdi": 0.40380423069000243,
"loss_ee": 0.5811583667993545,
"loss_delivery": 0.4012256935238838,
"loss_biodist": 0.12267061546444893,
"loss_toxic": 0.074160251673311
},
{
"loss": 2.0315637826919555,
"loss_size": 0.4700252488255501,
"loss_pdi": 0.41676153540611266,
"loss_ee": 0.5679270148277282,
"loss_delivery": 0.4009239628911018,
"loss_biodist": 0.13369966000318528,
"loss_toxic": 0.0645414180122316
},
{
"loss": 2.05439578294754,
"loss_size": 0.5344601511955261,
"loss_pdi": 0.40941277146339417,
"loss_ee": 0.5382379084825516,
"loss_delivery": 0.4047132283449173,
"loss_biodist": 0.13030270785093306,
"loss_toxic": 0.05972291301004588
},
{
"loss": 1.924245572090149,
"loss_size": 0.4832766890525818,
"loss_pdi": 0.3639328837394714,
"loss_ee": 0.5360181927680969,
"loss_delivery": 0.35968316793441774,
"loss_biodist": 0.12227168381214142,
"loss_toxic": 0.07759054847992956
}
],
"val": [
{
"loss": 212.32754516601562,
"loss_size": 0.3274073153734207,
"loss_pdi": 0.6179517060518265,
"loss_ee": 14.072934627532959,
"loss_delivery": 188.2798309326172,
"loss_biodist": 5.424441814422607,
"loss_toxic": 5.196369171142578,
"acc_pdi": 0.21212121212121213,
"acc_ee": 0.24242424242424243,
"acc_toxic": 0.9642857142857143
},
{
"loss": 75.10641860961914,
"loss_size": 0.8975031673908234,
"loss_pdi": 0.9911759197711945,
"loss_ee": 9.407026529312134,
"loss_delivery": 53.95008087158203,
"loss_biodist": 7.363304138183594,
"loss_toxic": 3.4896202087402344,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.24242424242424243,
"acc_toxic": 0.9642857142857143
},
{
"loss": 30.455476760864258,
"loss_size": 0.1754526048898697,
"loss_pdi": 2.747780442237854,
"loss_ee": 5.465596914291382,
"loss_delivery": 14.710479259490967,
"loss_biodist": 5.875157594680786,
"loss_toxic": 2.0271728634775172,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.24242424242424243,
"acc_toxic": 0.9642857142857143
},
{
"loss": 11.660218715667725,
"loss_size": 0.37073589861392975,
"loss_pdi": 1.9099485874176025,
"loss_ee": 2.634866714477539,
"loss_delivery": 3.382791042327881,
"loss_biodist": 2.98579603433609,
"loss_toxic": 0.600039035314694,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.24242424242424243,
"acc_toxic": 0.9642857142857143
},
{
"loss": 4.812026619911194,
"loss_size": 0.19489366561174393,
"loss_pdi": 1.1574044525623322,
"loss_ee": 1.303087830543518,
"loss_delivery": 0.7927329540252686,
"loss_biodist": 1.3637384474277496,
"loss_toxic": 0.07739773206412792,
"acc_pdi": 0.8484848484848485,
"acc_ee": 0.5151515151515151,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.9589792490005493,
"loss_size": 0.17401582933962345,
"loss_pdi": 1.0093275308609009,
"loss_ee": 0.7967433035373688,
"loss_delivery": 0.3617678307928145,
"loss_biodist": 0.5851162821054459,
"loss_toxic": 0.061497950460761786,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.860647201538086,
"loss_size": 0.16085178405046463,
"loss_pdi": 0.9414748549461365,
"loss_ee": 0.46526526287198067,
"loss_delivery": 0.5438784509897232,
"loss_biodist": 0.714705765247345,
"loss_toxic": 0.05823789437999949,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.7272727272727273,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.859875202178955,
"loss_size": 0.24519506096839905,
"loss_pdi": 0.9201228022575378,
"loss_ee": 0.6133346855640411,
"loss_delivery": 0.48798147588968277,
"loss_biodist": 0.5666100680828094,
"loss_toxic": 0.055127813189756125,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9642857142857143
},
{
"loss": 3.0383355617523193,
"loss_size": 0.19720425247214735,
"loss_pdi": 1.0486680269241333,
"loss_ee": 0.5307416841387749,
"loss_delivery": 0.7017943076789379,
"loss_biodist": 0.5510974824428558,
"loss_toxic": 0.044981934581301175,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.896167516708374,
"loss_size": 0.19404089637100697,
"loss_pdi": 0.9646035879850388,
"loss_ee": 0.5781851187348366,
"loss_delivery": 0.6214861165208276,
"loss_biodist": 0.5271554440259933,
"loss_toxic": 0.045089298400853295,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.851343035697937,
"loss_size": 0.20536951906979084,
"loss_pdi": 0.9235897362232208,
"loss_ee": 0.5952936001121998,
"loss_delivery": 0.7055458687245846,
"loss_biodist": 0.41286972165107727,
"loss_toxic": 0.04565286042634398,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.6704724431037903,
"loss_size": 0.18876394629478455,
"loss_pdi": 0.7482797801494598,
"loss_ee": 0.529376458376646,
"loss_delivery": 0.7572238328866661,
"loss_biodist": 0.4387248009443283,
"loss_toxic": 0.04034619980666321,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.554471731185913,
"loss_size": 0.15389276389032602,
"loss_pdi": 0.8669839203357697,
"loss_ee": 0.499736912548542,
"loss_delivery": 0.6517600808292627,
"loss_biodist": 0.3776693493127823,
"loss_toxic": 0.035502258921042085,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.7272727272727273,
"acc_toxic": 0.9642857142857143
},
{
"loss": 3.015945792198181,
"loss_size": 0.19041366316378117,
"loss_pdi": 1.0373757481575012,
"loss_ee": 0.533605795353651,
"loss_delivery": 0.8205330818891525,
"loss_biodist": 0.4419429898262024,
"loss_toxic": 0.03475822573818732,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.8732982873916626,
"loss_size": 0.1804293403401971,
"loss_pdi": 0.99785977602005,
"loss_ee": 0.5307499244809151,
"loss_delivery": 0.7448779745027423,
"loss_biodist": 0.42606888711452484,
"loss_toxic": 0.032939832817646675,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.8295711278915405,
"loss_size": 0.17214014939963818,
"loss_pdi": 0.9512898921966553,
"loss_ee": 0.5305828489363194,
"loss_delivery": 0.7462199348956347,
"loss_biodist": 0.4376784861087799,
"loss_toxic": 0.030546409892849624,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.826279878616333,
"loss_size": 0.1711592571809888,
"loss_pdi": 0.9499537348747253,
"loss_ee": 0.5237873047590256,
"loss_delivery": 0.7617753837257624,
"loss_biodist": 0.4285295158624649,
"loss_toxic": 0.03013451436709147,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.9642857142857143
},
{
"loss": 2.8353686332702637,
"loss_size": 0.17550351936370134,
"loss_pdi": 0.9350736439228058,
"loss_ee": 0.5329629778862,
"loss_delivery": 0.7670818213373423,
"loss_biodist": 0.4348946809768677,
"loss_toxic": 0.02934545233438257,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9642857142857143
}
]
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_3/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 4, 5, 7, 8, 9, 10, 11, 12, 14, 15, 16, 17, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 32, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 45, 46, 47, 51, 52, 53, 54, 55, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 73, 74, 75, 76, 77, 78, 79, 81, 82, 83, 84, 85, 86, 87, 89, 90, 91, 92, 93, 95, 96, 97, 98, 99, 100, 102, 104, 105, 106, 109, 110, 111, 112, 113, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 127, 129, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 146, 147, 148, 150, 151, 152, 153, 154, 155, 156, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 189, 190, 191, 192, 193, 194, 195, 198, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 212, 213, 214, 215, 216, 217, 218, 219, 221, 223, 224, 226, 227, 228, 230, 231, 233, 234, 235, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 251, 252, 253, 254, 255, 256, 257, 258, 259, 262, 263, 264, 265, 266, 267, 268, 269, 270, 272, 273, 274, 275, 276, 277, 278, 279, 280, 282, 284, 286, 288, 289, 290, 291, 292, 293, 294, 295, 297, 298, 299, 301, 302, 304, 305, 306, 307, 308, 309, 310, 312, 314, 315, 316, 317, 319, 320, 321, 325, 326, 327, 328, 331, 332, 334, 336, 340, 341, 343, 344, 346, 347, 348, 350, 352, 353, 357, 359, 360, 361, 362, 363, 364, 366, 367, 368, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 382, 383, 384, 385, 386, 387, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 400, 401, 402, 403, 404, 406, 407, 410, 411, 412, 414, 415, 416, 418, 419], "outer_test_idx": [3, 6, 13, 18, 19, 26, 33, 44, 48, 49, 50, 56, 57, 72, 80, 88, 94, 101, 103, 107, 108, 114, 126, 128, 130, 145, 149, 157, 167, 171, 175, 188, 196, 197, 199, 211, 220, 222, 225, 229, 232, 236, 250, 260, 261, 271, 281, 283, 285, 287, 296, 300, 303, 311, 313, 318, 322, 323, 324, 329, 330, 333, 335, 337, 338, 339, 342, 345, 349, 351, 354, 355, 356, 358, 365, 369, 381, 388, 399, 405, 408, 409, 413, 417]}

42
models/abl/baseline/20260623_205158/repeat_1/outer_fold_3/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 83,
"mse": 1.7722816318702095,
"rmse": 1.3312706831708605,
"mae": 0.5694137913208589,
"r2": -0.0024159991478140963
},
"delivery": {
"n_samples": 59,
"mse": 0.6715869625788514,
"rmse": 0.8195040955229275,
"mae": 0.48721023672654973,
"r2": 0.34713768051792715
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7023809523809523,
"precision": 0.6267796610169492,
"recall": 0.6370967741935484,
"f1": 0.6307367680675224
},
"ee": {
"n_samples": 84,
"accuracy": 0.6190476190476191,
"precision": 0.5535293250810492,
"recall": 0.5760683760683761,
"f1": 0.5530481924570595
},
"toxic": {
"n_samples": 60,
"accuracy": 0.9333333333333333,
"precision": 0.7142857142857143,
"recall": 0.9649122807017544,
"f1": 0.7818181818181817
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.27026190922231863,
"js_divergence": 0.06299181317830951
}
}

12
models/abl/baseline/20260623_205158/repeat_1/outer_fold_4/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.313123672812509,
"lr": 0.0009765192272158632,
"weight_decay": 0.0053740240978641905,
"backbone_lr_ratio": 0.13566938254683006,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_4/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 13}

279
models/abl/baseline/20260623_205158/repeat_1/outer_fold_4/history.json Normal file
View File

@ -0,0 +1,279 @@
{
"train": [
{
"loss": 5.3198872089385985,
"loss_size": 0.7633101761341095,
"loss_pdi": 0.69187433719635,
"loss_ee": 1.0832348108291625,
"loss_delivery": 1.0180094748735429,
"loss_biodist": 1.148764544725418,
"loss_toxic": 0.6217172026634217
},
{
"loss": 4.566192626953125,
"loss_size": 0.7208155035972595,
"loss_pdi": 0.6390170753002167,
"loss_ee": 1.0221041083335876,
"loss_delivery": 0.8898677706718445,
"loss_biodist": 0.7914427876472473,
"loss_toxic": 0.5198462009429932
},
{
"loss": 4.146983289718628,
"loss_size": 0.6776367217302323,
"loss_pdi": 0.6037499070167541,
"loss_ee": 0.9752871036529541,
"loss_delivery": 0.9565405771136284,
"loss_biodist": 0.5647945314645767,
"loss_toxic": 0.3892128989100456
},
{
"loss": 3.7967033624649047,
"loss_size": 0.7183691769838333,
"loss_pdi": 0.5778270542621613,
"loss_ee": 0.9329063177108765,
"loss_delivery": 0.8745826601982116,
"loss_biodist": 0.45337353348731996,
"loss_toxic": 0.2596608318388462
},
{
"loss": 3.3552826166152956,
"loss_size": 0.6089909583330154,
"loss_pdi": 0.5439370036125183,
"loss_ee": 0.8673183023929596,
"loss_delivery": 0.7894738852977753,
"loss_biodist": 0.35936565697193146,
"loss_toxic": 0.20247428640723228
},
{
"loss": 3.1958189487457274,
"loss_size": 0.5609132274985313,
"loss_pdi": 0.5376191437244415,
"loss_ee": 0.852169007062912,
"loss_delivery": 0.7843496188521385,
"loss_biodist": 0.3403240665793419,
"loss_toxic": 0.13841744996607303
},
{
"loss": 2.9731104135513307,
"loss_size": 0.49060665220022204,
"loss_pdi": 0.5260781496763229,
"loss_ee": 0.8076612710952759,
"loss_delivery": 0.7398313388228417,
"loss_biodist": 0.3146186485886574,
"loss_toxic": 0.11118286624550819
},
{
"loss": 2.8677444219589234,
"loss_size": 0.5340122729539871,
"loss_pdi": 0.5293333381414413,
"loss_ee": 0.7733309209346771,
"loss_delivery": 0.6874321639537812,
"loss_biodist": 0.28239639699459074,
"loss_toxic": 0.07821389734745025
},
{
"loss": 2.8695883989334106,
"loss_size": 0.456363120675087,
"loss_pdi": 0.48993122577667236,
"loss_ee": 0.7969828844070435,
"loss_delivery": 0.7760309278964996,
"loss_biodist": 0.25856100767850876,
"loss_toxic": 0.11154234390705824
},
{
"loss": 2.9714117527008055,
"loss_size": 0.4756880059838295,
"loss_pdi": 0.525236451625824,
"loss_ee": 0.7951141834259033,
"loss_delivery": 0.7354356780648231,
"loss_biodist": 0.2668692722916603,
"loss_toxic": 0.193408365547657
},
{
"loss": 2.769060969352722,
"loss_size": 0.4312173441052437,
"loss_pdi": 0.5284072905778885,
"loss_ee": 0.783972293138504,
"loss_delivery": 0.7113264828920365,
"loss_biodist": 0.2603357031941414,
"loss_toxic": 0.07603959087282419
},
{
"loss": 3.04667272567749,
"loss_size": 0.7416645839810372,
"loss_pdi": 0.5002941101789474,
"loss_ee": 0.7996781051158905,
"loss_delivery": 0.6856007933616638,
"loss_biodist": 0.24077045917510986,
"loss_toxic": 0.10171894459053873
},
{
"loss": 2.8001315355300904,
"loss_size": 0.45860919952392576,
"loss_pdi": 0.4763676881790161,
"loss_ee": 0.7997625172138214,
"loss_delivery": 0.742407999932766,
"loss_biodist": 0.23087860345840455,
"loss_toxic": 0.11384730339050293
}
],
"val": [
{
"loss": 151.37682723999023,
"loss_size": 109.16299438476562,
"loss_pdi": 0.8854561783373356,
"loss_ee": 13.746476173400879,
"loss_delivery": 24.054847717285156,
"loss_biodist": 2.788853645324707,
"loss_toxic": 1.333778738975525,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.24242424242424243,
"acc_toxic": 0.96
},
{
"loss": 94.67638397216797,
"loss_size": 79.8163070678711,
"loss_pdi": 0.630886547267437,
"loss_ee": 8.72434663772583,
"loss_delivery": 1.8503535985946655,
"loss_biodist": 2.0269339084625244,
"loss_toxic": 2.191328525543213,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.24242424242424243,
"acc_toxic": 0.96
},
{
"loss": 33.03701877593994,
"loss_size": 24.479528427124023,
"loss_pdi": 0.6195036470890045,
"loss_ee": 4.856629848480225,
"loss_delivery": 0.3994932472705841,
"loss_biodist": 1.5843472480773926,
"loss_toxic": 1.4051145315170288,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.24242424242424243,
"acc_toxic": 0.96
},
{
"loss": 7.850891351699829,
"loss_size": 4.216884076595306,
"loss_pdi": 0.656823456287384,
"loss_ee": 2.060563623905182,
"loss_delivery": 0.4691290855407715,
"loss_biodist": 0.4614642560482025,
"loss_toxic": 0.07764291763305664,
"acc_pdi": 0.6363636363636364,
"acc_ee": 0.36363636363636365,
"acc_toxic": 0.96
},
{
"loss": 4.38484525680542,
"loss_size": 1.9234002381563187,
"loss_pdi": 0.7618274986743927,
"loss_ee": 1.1175106167793274,
"loss_delivery": 0.3754430413246155,
"loss_biodist": 0.1797235757112503,
"loss_toxic": 0.0694519504904747,
"acc_pdi": 0.5454545454545454,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.96
},
{
"loss": 4.413366436958313,
"loss_size": 1.9679179787635803,
"loss_pdi": 0.7286995053291321,
"loss_ee": 1.1738770306110382,
"loss_delivery": 0.34018996357917786,
"loss_biodist": 0.1844538003206253,
"loss_toxic": 0.06463465094566345,
"acc_pdi": 0.6060606060606061,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.96
},
{
"loss": 4.314068794250488,
"loss_size": 1.865495353937149,
"loss_pdi": 0.7774328291416168,
"loss_ee": 1.105441153049469,
"loss_delivery": 0.34113433957099915,
"loss_biodist": 0.1971006989479065,
"loss_toxic": 0.07435857504606247,
"acc_pdi": 0.6363636363636364,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.96
},
{
"loss": 4.181123733520508,
"loss_size": 1.7796541452407837,
"loss_pdi": 0.7909885346889496,
"loss_ee": 1.0451889336109161,
"loss_delivery": 0.32920458912849426,
"loss_biodist": 0.19488513469696045,
"loss_toxic": 0.0862542912364006,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 0.96
},
{
"loss": 4.010855317115784,
"loss_size": 1.7509412914514542,
"loss_pdi": 0.7213941514492035,
"loss_ee": 0.9816495180130005,
"loss_delivery": 0.3286260962486267,
"loss_biodist": 0.18289732933044434,
"loss_toxic": 0.08557549864053726,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.96
},
{
"loss": 3.951658010482788,
"loss_size": 1.6630706042051315,
"loss_pdi": 0.7557414472103119,
"loss_ee": 1.01742422580719,
"loss_delivery": 0.31569284200668335,
"loss_biodist": 0.17840929329395294,
"loss_toxic": 0.06420498341321945,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.96
},
{
"loss": 3.9062973260879517,
"loss_size": 1.6621500700712204,
"loss_pdi": 0.7467974722385406,
"loss_ee": 0.9766978025436401,
"loss_delivery": 0.3118675649166107,
"loss_biodist": 0.18103832006454468,
"loss_toxic": 0.06865832954645157,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.96
},
{
"loss": 3.8756386041641235,
"loss_size": 1.641410455107689,
"loss_pdi": 0.751421332359314,
"loss_ee": 0.9626970589160919,
"loss_delivery": 0.3124241232872009,
"loss_biodist": 0.17649595439434052,
"loss_toxic": 0.07144398242235184,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.96
},
{
"loss": 3.8911858797073364,
"loss_size": 1.6369886845350266,
"loss_pdi": 0.7499736249446869,
"loss_ee": 0.9913561344146729,
"loss_delivery": 0.3116900622844696,
"loss_biodist": 0.1769104152917862,
"loss_toxic": 0.06587354838848114,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.96
}
]
}

1
models/abl/baseline/20260623_205158/repeat_1/outer_fold_4/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [1, 2, 3, 4, 6, 7, 8, 9, 10, 12, 13, 14, 15, 16, 17, 18, 19, 20, 22, 23, 25, 26, 28, 29, 31, 33, 34, 35, 37, 41, 42, 43, 44, 45, 46, 48, 49, 50, 51, 52, 53, 54, 56, 57, 59, 60, 62, 63, 64, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 77, 78, 79, 80, 81, 82, 83, 84, 85, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 101, 102, 103, 104, 105, 107, 108, 110, 111, 112, 113, 114, 116, 117, 119, 120, 121, 122, 123, 124, 126, 127, 128, 130, 132, 133, 134, 135, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 149, 152, 154, 155, 157, 160, 161, 162, 163, 164, 165, 166, 167, 169, 170, 171, 173, 174, 175, 176, 177, 178, 180, 181, 182, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 203, 205, 206, 207, 208, 209, 211, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 229, 230, 231, 232, 234, 235, 236, 238, 239, 240, 241, 243, 244, 245, 246, 247, 248, 249, 250, 251, 253, 254, 257, 258, 260, 261, 262, 263, 264, 265, 266, 267, 270, 271, 272, 273, 277, 278, 279, 280, 281, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 298, 299, 300, 301, 302, 303, 304, 305, 306, 309, 311, 312, 313, 314, 315, 316, 318, 319, 320, 322, 323, 324, 325, 326, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 341, 342, 343, 345, 346, 347, 349, 350, 351, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 365, 367, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 381, 383, 384, 385, 386, 388, 390, 391, 392, 393, 394, 395, 396, 397, 399, 400, 401, 402, 403, 405, 406, 407, 408, 409, 410, 411, 413, 415, 416, 417, 418, 419], "outer_test_idx": [0, 5, 11, 21, 24, 27, 30, 32, 36, 38, 39, 40, 47, 55, 58, 61, 65, 76, 86, 95, 100, 106, 109, 115, 118, 125, 129, 131, 136, 148, 150, 151, 153, 156, 158, 159, 168, 172, 179, 183, 184, 185, 202, 204, 210, 212, 213, 228, 233, 237, 242, 252, 255, 256, 259, 268, 269, 274, 275, 276, 282, 297, 307, 308, 310, 317, 321, 327, 340, 344, 348, 352, 364, 366, 368, 379, 380, 382, 387, 389, 398, 404, 412, 414]}

42
models/abl/baseline/20260623_205158/repeat_1/outer_fold_4/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.7536341327012797,
"rmse": 0.8681210357440255,
"mae": 0.5072707110472644,
"r2": -0.00845608290415334
},
"delivery": {
"n_samples": 58,
"mse": 1.087167994986309,
"rmse": 1.0426734843594656,
"mae": 0.6497257278654082,
"r2": -0.0512735040716632
},
"pdi": {
"n_samples": 84,
"accuracy": 0.75,
"precision": 0.6890838206627681,
"recall": 0.7133431085043989,
"f1": 0.6974789915966386
},
"ee": {
"n_samples": 84,
"accuracy": 0.6904761904761905,
"precision": 0.6308992939427722,
"recall": 0.6444444444444445,
"f1": 0.633663610371685
},
"toxic": {
"n_samples": 59,
"accuracy": 0.9661016949152542,
"precision": 0.8,
"recall": 0.9821428571428572,
"f1": 0.865909090909091
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.21472605299515538,
"js_divergence": 0.0586261381599948
}
}

12
models/abl/baseline/20260623_205158/repeat_2/outer_fold_0/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.18569939775458322,
"lr": 0.0009872295492725695,
"weight_decay": 0.017393516831905452,
"backbone_lr_ratio": 0.17239541174804215,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_0/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 11}

237
models/abl/baseline/20260623_205158/repeat_2/outer_fold_0/history.json Normal file
View File

@ -0,0 +1,237 @@
{
"train": [
{
"loss": 5.449092626571655,
"loss_size": 0.8463821470737457,
"loss_pdi": 0.6531818926334381,
"loss_ee": 1.0796023964881898,
"loss_delivery": 1.0442003339529038,
"loss_biodist": 1.142931294441223,
"loss_toxic": 0.6911491274833679
},
{
"loss": 4.678988337516785,
"loss_size": 1.0209883362054826,
"loss_pdi": 0.602011775970459,
"loss_ee": 0.9843448162078857,
"loss_delivery": 0.868944326043129,
"loss_biodist": 0.7855531632900238,
"loss_toxic": 0.43422114551067353
},
{
"loss": 3.8828047037124636,
"loss_size": 0.6727842763066292,
"loss_pdi": 0.5638415515422821,
"loss_ee": 0.9300254762172699,
"loss_delivery": 0.8391703262925148,
"loss_biodist": 0.568875539302826,
"loss_toxic": 0.32425700277090075
},
{
"loss": 3.950264048576355,
"loss_size": 1.0529354870319367,
"loss_pdi": 0.5390785008668899,
"loss_ee": 0.8994648933410645,
"loss_delivery": 0.7956141114234925,
"loss_biodist": 0.4504675894975662,
"loss_toxic": 0.2338446371257305
},
{
"loss": 3.171139967441559,
"loss_size": 0.67209033370018,
"loss_pdi": 0.5058561980724334,
"loss_ee": 0.7969049036502838,
"loss_delivery": 0.7200598813593387,
"loss_biodist": 0.35736297369003295,
"loss_toxic": 0.13218607269227506
},
{
"loss": 3.089067244529724,
"loss_size": 0.5777189195156097,
"loss_pdi": 0.45978887677192687,
"loss_ee": 0.7973751246929168,
"loss_delivery": 0.7926068544387818,
"loss_biodist": 0.33243380635976794,
"loss_toxic": 0.14422718472778798
},
{
"loss": 2.824042725563049,
"loss_size": 0.5082829475402832,
"loss_pdi": 0.46822271645069125,
"loss_ee": 0.7749077081680298,
"loss_delivery": 0.7078689366579056,
"loss_biodist": 0.27962898164987565,
"loss_toxic": 0.09844160005450249
},
{
"loss": 2.8571096420288087,
"loss_size": 0.4909460961818695,
"loss_pdi": 0.4669936686754227,
"loss_ee": 0.7580368220806122,
"loss_delivery": 0.7777129769325256,
"loss_biodist": 0.2535310655832291,
"loss_toxic": 0.12509770039469004
},
{
"loss": 2.6829344511032103,
"loss_size": 0.4839496031403542,
"loss_pdi": 0.4512131690979004,
"loss_ee": 0.743899917602539,
"loss_delivery": 0.6636333405971527,
"loss_biodist": 0.240355783700943,
"loss_toxic": 0.11239960938692092
},
{
"loss": 2.8401250004768372,
"loss_size": 0.679023240506649,
"loss_pdi": 0.44916839003562925,
"loss_ee": 0.7359160959720612,
"loss_delivery": 0.6693838536739349,
"loss_biodist": 0.2336220309138298,
"loss_toxic": 0.08850220618769526
},
{
"loss": 2.531557321548462,
"loss_size": 0.4534479148685932,
"loss_pdi": 0.4408748745918274,
"loss_ee": 0.7259799718856812,
"loss_delivery": 0.6009897664189339,
"loss_biodist": 0.23776510804891587,
"loss_toxic": 0.0835730129852891
}
],
"val": [
{
"loss": 188.90975952148438,
"loss_size": 45.01241683959961,
"loss_pdi": 3.2897791862487793,
"loss_ee": 23.29022216796875,
"loss_delivery": 103.00629806518555,
"loss_biodist": 14.998572826385498,
"loss_toxic": 0.5866744741797447,
"acc_pdi": 0.24242424242424243,
"acc_ee": 0.15151515151515152,
"acc_toxic": 0.9583333333333334
},
{
"loss": 65.91048431396484,
"loss_size": 17.505257606506348,
"loss_pdi": 0.6360184773802757,
"loss_ee": 13.60608434677124,
"loss_delivery": 18.747560024261475,
"loss_biodist": 14.360416889190674,
"loss_toxic": 1.8936293558363104,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.15151515151515152,
"acc_toxic": 0.9583333333333334
},
{
"loss": 20.623906135559082,
"loss_size": 1.8114935755729675,
"loss_pdi": 0.772331653162837,
"loss_ee": 6.966356515884399,
"loss_delivery": 1.8137834966182709,
"loss_biodist": 7.9090540409088135,
"loss_toxic": 1.7319070062949322,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.15151515151515152,
"acc_toxic": 0.9583333333333334
},
{
"loss": 3.71417498588562,
"loss_size": 0.3245811965316534,
"loss_pdi": 0.6202343385666609,
"loss_ee": 1.3963280022144318,
"loss_delivery": 0.5899864500388503,
"loss_biodist": 0.6940107941627502,
"loss_toxic": 0.12449116259813309,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.30303030303030304,
"acc_toxic": 0.9583333333333334
},
{
"loss": 1.81367789208889,
"loss_size": 0.23402975732460618,
"loss_pdi": 0.29450613632798195,
"loss_ee": 0.4783802628517151,
"loss_delivery": 0.45590384281240404,
"loss_biodist": 0.2778492271900177,
"loss_toxic": 0.05870048888027668,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9583333333333334
},
{
"loss": 1.8064501881599426,
"loss_size": 0.24365851655602455,
"loss_pdi": 0.30049131717532873,
"loss_ee": 0.4604756459593773,
"loss_delivery": 0.408929499797523,
"loss_biodist": 0.3284158557653427,
"loss_toxic": 0.051831915974617004,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9583333333333334
},
{
"loss": 1.8169548511505127,
"loss_size": 0.24683533981442451,
"loss_pdi": 0.30905733723193407,
"loss_ee": 0.44691873900592327,
"loss_delivery": 0.38278661756521615,
"loss_biodist": 0.36872436106204987,
"loss_toxic": 0.05168247036635876,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9583333333333334
},
{
"loss": 1.8818739354610443,
"loss_size": 0.26859522610902786,
"loss_pdi": 0.30615095095708966,
"loss_ee": 0.4428381510078907,
"loss_delivery": 0.3961277920752764,
"loss_biodist": 0.4104864299297333,
"loss_toxic": 0.049348259344697,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.9583333333333334
},
{
"loss": 1.8650652170181274,
"loss_size": 0.2776916176080704,
"loss_pdi": 0.29989287769421935,
"loss_ee": 0.44173157401382923,
"loss_delivery": 0.3812694759108126,
"loss_biodist": 0.4180755913257599,
"loss_toxic": 0.03872741665691137,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9583333333333334
},
{
"loss": 1.860511988401413,
"loss_size": 0.2812216803431511,
"loss_pdi": 0.30004148418083787,
"loss_ee": 0.44041919335722923,
"loss_delivery": 0.3759109512902796,
"loss_biodist": 0.4222371429204941,
"loss_toxic": 0.03345674928277731,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.9583333333333334
},
{
"loss": 1.8582188487052917,
"loss_size": 0.29131920635700226,
"loss_pdi": 0.29559751553460956,
"loss_ee": 0.4416970517486334,
"loss_delivery": 0.37560337950708345,
"loss_biodist": 0.40974242985248566,
"loss_toxic": 0.03689831122756004,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.9583333333333334
}
]
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_0/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 21, 22, 23, 24, 25, 26, 27, 28, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 52, 53, 54, 55, 56, 57, 58, 61, 62, 63, 64, 65, 66, 67, 68, 71, 72, 73, 75, 76, 77, 80, 81, 85, 86, 88, 89, 90, 92, 94, 95, 96, 98, 99, 100, 101, 103, 104, 105, 106, 107, 108, 109, 111, 113, 114, 115, 116, 117, 118, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 133, 134, 135, 136, 138, 139, 141, 144, 145, 146, 147, 148, 149, 150, 151, 153, 154, 155, 156, 157, 158, 159, 160, 163, 164, 167, 168, 169, 170, 171, 172, 175, 176, 178, 179, 180, 182, 183, 184, 185, 188, 189, 190, 191, 192, 193, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 206, 208, 210, 211, 212, 213, 216, 217, 218, 219, 220, 222, 224, 225, 227, 228, 229, 230, 231, 232, 233, 234, 236, 237, 239, 241, 242, 244, 245, 247, 248, 249, 250, 252, 253, 255, 256, 257, 258, 259, 260, 261, 262, 265, 267, 268, 269, 270, 271, 272, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 285, 286, 287, 288, 291, 292, 294, 296, 297, 298, 299, 300, 301, 303, 304, 306, 307, 308, 309, 310, 311, 312, 313, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 332, 333, 335, 336, 337, 338, 339, 340, 342, 343, 344, 345, 346, 347, 348, 349, 351, 352, 354, 355, 356, 357, 358, 359, 360, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 375, 377, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 393, 395, 396, 397, 398, 399, 400, 401, 403, 404, 405, 406, 407, 408, 409, 411, 412, 413, 414, 415, 417, 418, 419], "outer_test_idx": [7, 16, 20, 29, 51, 59, 60, 69, 70, 74, 78, 79, 82, 83, 84, 87, 91, 93, 97, 102, 110, 112, 119, 120, 121, 132, 137, 140, 142, 143, 152, 161, 162, 165, 166, 173, 174, 177, 181, 186, 187, 194, 205, 207, 209, 214, 215, 221, 223, 226, 235, 238, 240, 243, 246, 251, 254, 263, 264, 266, 273, 284, 289, 290, 293, 295, 302, 305, 314, 315, 331, 334, 341, 350, 353, 361, 373, 376, 378, 392, 394, 402, 410, 416]}

42
models/abl/baseline/20260623_205158/repeat_2/outer_fold_0/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 83,
"mse": 1.5072429789639405,
"rmse": 1.2276982442619768,
"mae": 0.5558247003165713,
"r2": 0.07753019623566748
},
"delivery": {
"n_samples": 58,
"mse": 0.5835939668070607,
"rmse": 0.76393322143173,
"mae": 0.5947980444942569,
"r2": 0.2582168518421626
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7738095238095238,
"precision": 0.7232142857142857,
"recall": 0.7494654312188168,
"f1": 0.7325289089994973
},
"ee": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6729142936039488,
"recall": 0.7214285714285715,
"f1": 0.6806141997028696
},
"toxic": {
"n_samples": 58,
"accuracy": 0.9655172413793104,
"precision": 0.75,
"recall": 0.9821428571428572,
"f1": 0.8242424242424242
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.28982565124202386,
"js_divergence": 0.06069208833134297
}
}

12
models/abl/baseline/20260623_205158/repeat_2/outer_fold_1/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.36640960296080094,
"lr": 0.0009935032389272061,
"weight_decay": 0.007506834667713186,
"backbone_lr_ratio": 0.10108281143505408,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_1/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 11}

237
models/abl/baseline/20260623_205158/repeat_2/outer_fold_1/history.json Normal file
View File

@ -0,0 +1,237 @@
{
"train": [
{
"loss": 5.901606559753418,
"loss_size": 1.2065881758928299,
"loss_pdi": 0.6568825721740723,
"loss_ee": 1.0730257749557495,
"loss_delivery": 1.1219383984804154,
"loss_biodist": 1.1867228746414185,
"loss_toxic": 0.6668074429035187
},
{
"loss": 4.969041323661804,
"loss_size": 1.1804757505655288,
"loss_pdi": 0.6160871386528015,
"loss_ee": 0.9857695698738098,
"loss_delivery": 0.8546574175357818,
"loss_biodist": 0.8020791649818421,
"loss_toxic": 0.550473666191101
},
{
"loss": 4.529213881492614,
"loss_size": 1.135186380147934,
"loss_pdi": 0.595760029554367,
"loss_ee": 0.9603484511375427,
"loss_delivery": 0.8062571913003922,
"loss_biodist": 0.5837476968765258,
"loss_toxic": 0.4725710928440094
},
{
"loss": 4.098095178604126,
"loss_size": 1.0632753282785417,
"loss_pdi": 0.5697092026472091,
"loss_ee": 0.8684752643108368,
"loss_delivery": 0.8286568135023117,
"loss_biodist": 0.4417699038982391,
"loss_toxic": 0.3501173913478851
},
{
"loss": 3.7869781255722046,
"loss_size": 0.9929396659135818,
"loss_pdi": 0.5613552153110504,
"loss_ee": 0.8111471951007843,
"loss_delivery": 0.7437609761953354,
"loss_biodist": 0.4380580335855484,
"loss_toxic": 0.26053340286016463
},
{
"loss": 3.6553616285324098,
"loss_size": 0.918002250790596,
"loss_pdi": 0.541991326212883,
"loss_ee": 0.8302526533603668,
"loss_delivery": 0.7477440476417542,
"loss_biodist": 0.3995376527309418,
"loss_toxic": 0.23948886543512343
},
{
"loss": 3.521389031410217,
"loss_size": 0.9813198298215866,
"loss_pdi": 0.5267888069152832,
"loss_ee": 0.7885477840900421,
"loss_delivery": 0.6853216931223869,
"loss_biodist": 0.337273308634758,
"loss_toxic": 0.22192156538367272
},
{
"loss": 3.2853312730789184,
"loss_size": 0.8076761543750763,
"loss_pdi": 0.5233874380588531,
"loss_ee": 0.776021021604538,
"loss_delivery": 0.7353604823350907,
"loss_biodist": 0.3144132807850838,
"loss_toxic": 0.1465326279401779
},
{
"loss": 3.3931561946868896,
"loss_size": 0.8530812963843346,
"loss_pdi": 0.5077527463436127,
"loss_ee": 0.7840090811252594,
"loss_delivery": 0.7130382150411606,
"loss_biodist": 0.3390728488564491,
"loss_toxic": 0.21623777076601983
},
{
"loss": 3.408563423156738,
"loss_size": 0.9367903709411621,
"loss_pdi": 0.5393963605165482,
"loss_ee": 0.7679374992847443,
"loss_delivery": 0.7398537293076515,
"loss_biodist": 0.2991093099117279,
"loss_toxic": 0.14827398546040058
},
{
"loss": 3.3290008068084718,
"loss_size": 0.8624186828732491,
"loss_pdi": 0.5163982570171356,
"loss_ee": 0.7700368523597717,
"loss_delivery": 0.7174924358725547,
"loss_biodist": 0.34391860365867616,
"loss_toxic": 0.13990672975778579
}
],
"val": [
{
"loss": 132.56271362304688,
"loss_size": 81.33299255371094,
"loss_pdi": 2.5050874351491075,
"loss_ee": 7.608971118927002,
"loss_delivery": 20.74916648864746,
"loss_biodist": 20.621540546417236,
"loss_toxic": 0.6091575622558594,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.2727272727272727,
"acc_toxic": 1.0
},
{
"loss": 65.57118225097656,
"loss_size": 35.65294075012207,
"loss_pdi": 2.0960091285814997,
"loss_ee": 5.895962297915389,
"loss_delivery": 3.148690491914749,
"loss_biodist": 19.59813117980957,
"loss_toxic": 0.016315508168190718,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.2727272727272727,
"acc_toxic": 1.0
},
{
"loss": 26.14203453063965,
"loss_size": 8.701539516448975,
"loss_pdi": 1.472307329531759,
"loss_ee": 3.043973806896247,
"loss_delivery": 1.5724104046821594,
"loss_biodist": 11.786346197128296,
"loss_toxic": 0.026583264581859112,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.2727272727272727,
"acc_toxic": 1.0
},
{
"loss": 7.222604513168335,
"loss_size": 1.2069462537765503,
"loss_pdi": 0.7914322391152382,
"loss_ee": 1.4730214625597,
"loss_delivery": 1.128986656665802,
"loss_biodist": 2.507304310798645,
"loss_toxic": 0.22003231942653656,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.3939393939393939,
"acc_toxic": 0.9565217391304348
},
{
"loss": 3.6449748277664185,
"loss_size": 0.6141378283500671,
"loss_pdi": 0.45458294451236725,
"loss_ee": 1.5160790383815765,
"loss_delivery": 0.47068952023983,
"loss_biodist": 0.4285694509744644,
"loss_toxic": 0.19025897607207298,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9565217391304348
},
{
"loss": 3.7185298204421997,
"loss_size": 0.6736274808645248,
"loss_pdi": 0.44979293644428253,
"loss_ee": 1.556200534105301,
"loss_delivery": 0.5530323311686516,
"loss_biodist": 0.38627544045448303,
"loss_toxic": 0.1352948360145092,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9565217391304348
},
{
"loss": 3.745786190032959,
"loss_size": 0.658877432346344,
"loss_pdi": 0.45660917460918427,
"loss_ee": 1.6202260553836823,
"loss_delivery": 0.5536150559782982,
"loss_biodist": 0.3847600221633911,
"loss_toxic": 0.11236662603914738,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9565217391304348
},
{
"loss": 3.81522536277771,
"loss_size": 0.6509192734956741,
"loss_pdi": 0.47273319959640503,
"loss_ee": 1.6899810135364532,
"loss_delivery": 0.4969055950641632,
"loss_biodist": 0.454554483294487,
"loss_toxic": 0.09597210586071014,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9565217391304348
},
{
"loss": 3.8280935287475586,
"loss_size": 0.631020724773407,
"loss_pdi": 0.4882955402135849,
"loss_ee": 1.7266646325588226,
"loss_delivery": 0.46563348919153214,
"loss_biodist": 0.47606857120990753,
"loss_toxic": 0.08867285028100014,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9565217391304348
},
{
"loss": 3.791571617126465,
"loss_size": 0.617971658706665,
"loss_pdi": 0.4741324335336685,
"loss_ee": 1.7368178069591522,
"loss_delivery": 0.46720805019140244,
"loss_biodist": 0.4530954211950302,
"loss_toxic": 0.09044981561601162,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9565217391304348
},
{
"loss": 3.786234140396118,
"loss_size": 0.6090261340141296,
"loss_pdi": 0.47116221487522125,
"loss_ee": 1.7393423318862915,
"loss_delivery": 0.47243160009384155,
"loss_biodist": 0.4540674239397049,
"loss_toxic": 0.0884552150964737,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.9565217391304348
}
]
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_1/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 3, 5, 6, 7, 10, 11, 12, 13, 15, 16, 18, 19, 20, 21, 24, 26, 27, 29, 30, 31, 32, 33, 35, 36, 38, 39, 40, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 65, 66, 67, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 97, 98, 99, 100, 101, 102, 103, 105, 106, 107, 108, 109, 110, 112, 114, 115, 118, 119, 120, 121, 123, 124, 125, 126, 128, 129, 130, 131, 132, 136, 137, 140, 141, 142, 143, 144, 145, 146, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 161, 162, 164, 165, 166, 167, 168, 169, 171, 172, 173, 174, 175, 176, 177, 178, 179, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 194, 195, 196, 197, 198, 199, 201, 202, 204, 205, 206, 207, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 225, 226, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 240, 242, 243, 244, 246, 247, 248, 250, 251, 252, 254, 255, 256, 258, 259, 260, 261, 262, 263, 264, 266, 268, 269, 270, 271, 273, 274, 275, 276, 277, 278, 279, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 295, 296, 297, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 313, 314, 315, 316, 317, 318, 319, 321, 322, 323, 324, 325, 327, 328, 329, 330, 331, 333, 334, 335, 337, 338, 339, 340, 341, 342, 343, 344, 345, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 361, 362, 364, 365, 366, 367, 368, 369, 373, 374, 375, 376, 378, 379, 380, 381, 382, 384, 387, 388, 389, 391, 392, 393, 394, 396, 397, 398, 399, 402, 403, 404, 405, 406, 408, 409, 410, 411, 412, 413, 414, 416, 417, 419], "outer_test_idx": [1, 2, 4, 8, 9, 14, 17, 22, 23, 25, 28, 34, 37, 41, 43, 45, 63, 64, 68, 85, 96, 104, 111, 113, 116, 117, 122, 127, 133, 134, 135, 138, 139, 147, 160, 163, 170, 180, 191, 192, 193, 200, 203, 208, 224, 227, 230, 239, 241, 245, 249, 253, 257, 265, 267, 272, 280, 294, 298, 299, 309, 312, 320, 326, 332, 336, 346, 347, 360, 363, 370, 371, 372, 377, 383, 385, 386, 390, 395, 400, 401, 407, 415, 418]}

42
models/abl/baseline/20260623_205158/repeat_2/outer_fold_1/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.4057494802572597,
"rmse": 0.6369846781966264,
"mae": 0.5019206670778138,
"r2": -0.2390146595647713
},
"delivery": {
"n_samples": 61,
"mse": 1.095192634861098,
"rmse": 1.0465145172720243,
"mae": 0.6369023019234177,
"r2": 0.17101812667726723
},
"pdi": {
"n_samples": 84,
"accuracy": 0.6190476190476191,
"precision": 0.5575167376749848,
"recall": 0.5714285714285714,
"f1": 0.5543766578249337
},
"ee": {
"n_samples": 84,
"accuracy": 0.5714285714285714,
"precision": 0.5153903903903904,
"recall": 0.5209076637648066,
"f1": 0.507505844715147
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9508196721311475,
"precision": 0.75,
"recall": 0.9741379310344828,
"f1": 0.8200589970501475
},
"biodist": {
"n_samples": 61,
"kl_divergence": 0.3834603232909285,
"js_divergence": 0.0914682060621546
}
}

12
models/abl/baseline/20260623_205158/repeat_2/outer_fold_2/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.318342405407684,
"lr": 0.0008937362022799393,
"weight_decay": 0.06633867694372529,
"backbone_lr_ratio": 0.20906844321970927,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_2/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 10}

216
models/abl/baseline/20260623_205158/repeat_2/outer_fold_2/history.json Normal file
View File

@ -0,0 +1,216 @@
{
"train": [
{
"loss": 5.740620565414429,
"loss_size": 1.1896785780787469,
"loss_pdi": 0.6869230091571807,
"loss_ee": 1.0699448347091676,
"loss_delivery": 1.03231680393219,
"loss_biodist": 1.136617010831833,
"loss_toxic": 0.6333471000194549
},
{
"loss": 5.005722069740296,
"loss_size": 1.1524270862340926,
"loss_pdi": 0.6445027768611908,
"loss_ee": 1.0133706212043763,
"loss_delivery": 0.8842641621828079,
"loss_biodist": 0.8097323775291443,
"loss_toxic": 0.5201924949884414
},
{
"loss": 4.6578610897064205,
"loss_size": 1.1203607082366944,
"loss_pdi": 0.6211460649967193,
"loss_ee": 0.9555845499038697,
"loss_delivery": 0.8929742038249969,
"loss_biodist": 0.6400978416204453,
"loss_toxic": 0.45236344933509826
},
{
"loss": 4.149107980728149,
"loss_size": 1.0351139575242996,
"loss_pdi": 0.60948166847229,
"loss_ee": 0.8915052592754364,
"loss_delivery": 0.8007120192050934,
"loss_biodist": 0.49783630669116974,
"loss_toxic": 0.338179911673069
},
{
"loss": 4.084399080276489,
"loss_size": 1.2795856654644013,
"loss_pdi": 0.5381818920373916,
"loss_ee": 0.8509327173233032,
"loss_delivery": 0.831762258708477,
"loss_biodist": 0.41192333698272704,
"loss_toxic": 0.1975079134106636
},
{
"loss": 3.8901400566101074,
"loss_size": 1.2531759902834891,
"loss_pdi": 0.5502055495977402,
"loss_ee": 0.808403742313385,
"loss_delivery": 0.7362294673919678,
"loss_biodist": 0.3942125141620636,
"loss_toxic": 0.1723215252161026
},
{
"loss": 3.4875223636627197,
"loss_size": 0.9208602800965309,
"loss_pdi": 0.5349549561738968,
"loss_ee": 0.8125870168209076,
"loss_delivery": 0.7612511485815048,
"loss_biodist": 0.3403301119804382,
"loss_toxic": 0.13829674795269967
},
{
"loss": 3.278315043449402,
"loss_size": 0.8509392693638802,
"loss_pdi": 0.5423421293497086,
"loss_ee": 0.7730867922306061,
"loss_delivery": 0.6735098049044609,
"loss_biodist": 0.34112584590911865,
"loss_toxic": 0.11526295617222786
},
{
"loss": 3.3390276193618775,
"loss_size": 0.8696594685316086,
"loss_pdi": 0.5174867331981658,
"loss_ee": 0.7572001457214356,
"loss_delivery": 0.7882219508290291,
"loss_biodist": 0.31793602705001833,
"loss_toxic": 0.10892055258154869
},
{
"loss": 3.4321290254592896,
"loss_size": 1.0273638755083083,
"loss_pdi": 0.5237745851278305,
"loss_ee": 0.7504259288311005,
"loss_delivery": 0.7083810180425644,
"loss_biodist": 0.3297483652830124,
"loss_toxic": 0.11391140334308147
}
],
"val": [
{
"loss": 231.37726593017578,
"loss_size": 9.375548362731934,
"loss_pdi": 0.5674893781542778,
"loss_ee": 5.383962273597703,
"loss_delivery": 210.97039031982422,
"loss_biodist": 4.298196315765381,
"loss_toxic": 2.5510331686527934,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.36363636363636365,
"acc_toxic": 0.92
},
{
"loss": 63.84561538696289,
"loss_size": 3.781942844390869,
"loss_pdi": 0.6476910635828972,
"loss_ee": 3.4568774700001086,
"loss_delivery": 48.18235397338867,
"loss_biodist": 5.315637171268463,
"loss_toxic": 3.3002430795677355,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.36363636363636365,
"acc_toxic": 0.92
},
{
"loss": 17.31868553161621,
"loss_size": 0.17151660099625587,
"loss_pdi": 0.6472159698605537,
"loss_ee": 2.024239696416771,
"loss_delivery": 9.08073115348816,
"loss_biodist": 3.3038226664066315,
"loss_toxic": 2.3683474029166973,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.36363636363636365,
"acc_toxic": 0.92
},
{
"loss": 4.6316773891448975,
"loss_size": 0.2273484580218792,
"loss_pdi": 0.48698216304183006,
"loss_ee": 0.7906219884753227,
"loss_delivery": 1.8861270546913147,
"loss_biodist": 0.8904538005590439,
"loss_toxic": 0.3924381013493985,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.36363636363636365,
"acc_toxic": 0.92
},
{
"loss": 2.274237334728241,
"loss_size": 0.19311259221285582,
"loss_pdi": 0.44868385791778564,
"loss_ee": 0.716453343629837,
"loss_delivery": 0.6562463492155075,
"loss_biodist": 0.17407863587141037,
"loss_toxic": 0.07998653873801231,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.96
},
{
"loss": 2.4465495347976685,
"loss_size": 0.24319320544600487,
"loss_pdi": 0.542385071516037,
"loss_ee": 0.7102380990982056,
"loss_delivery": 0.6364419311285019,
"loss_biodist": 0.2279042899608612,
"loss_toxic": 0.08691083639860153,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.96
},
{
"loss": 2.4209267497062683,
"loss_size": 0.2700926959514618,
"loss_pdi": 0.5215522944927216,
"loss_ee": 0.7110106945037842,
"loss_delivery": 0.5964515656232834,
"loss_biodist": 0.2424968034029007,
"loss_toxic": 0.08071399852633476,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.96
},
{
"loss": 2.4743611216545105,
"loss_size": 0.2518749479204416,
"loss_pdi": 0.5164230465888977,
"loss_ee": 0.7619437575340271,
"loss_delivery": 0.6075122505426407,
"loss_biodist": 0.2569751664996147,
"loss_toxic": 0.08403769135475159,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.96
},
{
"loss": 2.409421384334564,
"loss_size": 0.2385507132858038,
"loss_pdi": 0.509266272187233,
"loss_ee": 0.7387356460094452,
"loss_delivery": 0.6034215986728668,
"loss_biodist": 0.2423449456691742,
"loss_toxic": 0.08040330559015274,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.96
},
{
"loss": 2.4236229062080383,
"loss_size": 0.23541532177478075,
"loss_pdi": 0.5067342072725296,
"loss_ee": 0.7435013651847839,
"loss_delivery": 0.6098943501710892,
"loss_biodist": 0.24742697924375534,
"loss_toxic": 0.08437797799706459,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.96
}
]
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_2/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 13, 14, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 32, 33, 34, 36, 37, 38, 39, 40, 41, 43, 44, 45, 47, 48, 49, 50, 51, 55, 56, 57, 58, 59, 60, 61, 63, 64, 65, 68, 69, 70, 72, 74, 76, 78, 79, 80, 82, 83, 84, 85, 86, 87, 88, 91, 93, 94, 95, 96, 97, 100, 101, 102, 103, 104, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 142, 143, 145, 147, 148, 149, 150, 151, 152, 153, 156, 157, 158, 159, 160, 161, 162, 163, 165, 166, 167, 168, 170, 171, 172, 173, 174, 175, 177, 179, 180, 181, 183, 184, 185, 186, 187, 188, 191, 192, 193, 194, 196, 197, 199, 200, 202, 203, 204, 205, 207, 208, 209, 210, 211, 212, 213, 214, 215, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 232, 233, 235, 236, 237, 238, 239, 240, 241, 242, 243, 245, 246, 249, 250, 251, 252, 253, 254, 255, 256, 257, 259, 260, 261, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 280, 281, 282, 283, 284, 285, 287, 289, 290, 293, 294, 295, 296, 297, 298, 299, 300, 302, 303, 305, 307, 308, 309, 310, 311, 312, 313, 314, 315, 317, 318, 320, 321, 322, 323, 324, 326, 327, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 358, 360, 361, 363, 364, 365, 366, 368, 369, 370, 371, 372, 373, 376, 377, 378, 379, 380, 381, 382, 383, 385, 386, 387, 388, 389, 390, 392, 394, 395, 398, 399, 400, 401, 402, 404, 405, 407, 408, 409, 410, 412, 413, 414, 415, 416, 417, 418], "outer_test_idx": [10, 12, 15, 31, 35, 42, 46, 52, 53, 54, 62, 66, 67, 71, 73, 75, 77, 81, 89, 90, 92, 98, 99, 105, 123, 124, 141, 144, 146, 154, 155, 164, 169, 176, 178, 182, 189, 190, 195, 198, 201, 206, 216, 217, 218, 219, 231, 234, 244, 247, 248, 258, 262, 270, 277, 278, 279, 286, 288, 291, 292, 301, 304, 306, 316, 319, 325, 328, 343, 357, 359, 362, 367, 374, 375, 384, 391, 393, 396, 397, 403, 406, 411, 419]}

42
models/abl/baseline/20260623_205158/repeat_2/outer_fold_2/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.4667056344892945,
"rmse": 0.6831585719943024,
"mae": 0.5159196099488154,
"r2": 0.07826471049783545
},
"delivery": {
"n_samples": 60,
"mse": 0.6618025560042317,
"rmse": 0.8135124805460772,
"mae": 0.5426560255466029,
"r2": 0.11902609175763024
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6517857142857143,
"recall": 0.6744868035190617,
"f1": 0.6583050847457628
},
"ee": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.6138888888888889,
"recall": 0.6785714285714285,
"f1": 0.6291768416768416
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9344262295081968,
"precision": 0.7142857142857143,
"recall": 0.9655172413793103,
"f1": 0.7821428571428571
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.3327396562359482,
"js_divergence": 0.08480501673906925
}
}

12
models/abl/baseline/20260623_205158/repeat_2/outer_fold_3/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.12612237693012526,
"lr": 0.0009464809649969671,
"weight_decay": 0.07051438027639334,
"backbone_lr_ratio": 0.9343703462702573,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_3/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 18}

382
models/abl/baseline/20260623_205158/repeat_2/outer_fold_3/history.json Normal file
View File

@ -0,0 +1,382 @@
{
"train": [
{
"loss": 5.324610257148743,
"loss_size": 0.8198839724063873,
"loss_pdi": 0.6737089335918427,
"loss_ee": 1.0780839800834656,
"loss_delivery": 0.9609979882836341,
"loss_biodist": 1.1395575404167175,
"loss_toxic": 0.6576838761568069
},
{
"loss": 4.3447551965713505,
"loss_size": 0.7436282724142075,
"loss_pdi": 0.6086643695831299,
"loss_ee": 0.9876380443572998,
"loss_delivery": 0.7780480682849884,
"loss_biodist": 0.776026827096939,
"loss_toxic": 0.46212508380413053
},
{
"loss": 3.7673992395401,
"loss_size": 0.6659511342644692,
"loss_pdi": 0.5844933927059174,
"loss_ee": 0.8911072075366974,
"loss_delivery": 0.7604616850614547,
"loss_biodist": 0.5500574856996536,
"loss_toxic": 0.3273125126957893
},
{
"loss": 4.057488012313843,
"loss_size": 0.8315710723400116,
"loss_pdi": 0.6230540633201599,
"loss_ee": 0.9625594437122345,
"loss_delivery": 0.8757962495088577,
"loss_biodist": 0.5533048868179321,
"loss_toxic": 0.23461032062768936
},
{
"loss": 3.6338284969329835,
"loss_size": 0.7274438977241516,
"loss_pdi": 0.5866671562194824,
"loss_ee": 0.9285650968551635,
"loss_delivery": 0.7937903434038163,
"loss_biodist": 0.44598024487495425,
"loss_toxic": 0.1745324343442917
},
{
"loss": 3.266851019859314,
"loss_size": 0.7233955949544907,
"loss_pdi": 0.54100341796875,
"loss_ee": 0.8078328907489777,
"loss_delivery": 0.7360958337783814,
"loss_biodist": 0.3374472841620445,
"loss_toxic": 0.1397423676593462
},
{
"loss": 3.5112888336181642,
"loss_size": 1.0364958122372627,
"loss_pdi": 0.48492484986782075,
"loss_ee": 0.7834768772125245,
"loss_delivery": 0.7065127834677696,
"loss_biodist": 0.2921538516879082,
"loss_toxic": 0.22646510936319827
},
{
"loss": 3.0539005517959597,
"loss_size": 0.6716255933046341,
"loss_pdi": 0.5070259064435959,
"loss_ee": 0.766530328989029,
"loss_delivery": 0.6904316738247871,
"loss_biodist": 0.28585429936647416,
"loss_toxic": 0.15391857847571372
},
{
"loss": 2.941991186141968,
"loss_size": 0.6921013623476029,
"loss_pdi": 0.4792163670063019,
"loss_ee": 0.7757922112941742,
"loss_delivery": 0.6609947487711907,
"loss_biodist": 0.25257839187979697,
"loss_toxic": 0.10671633686870337
},
{
"loss": 2.6624006032943726,
"loss_size": 0.670102259516716,
"loss_pdi": 0.48275184333324433,
"loss_ee": 0.6530667543411255,
"loss_delivery": 0.55238878428936,
"loss_biodist": 0.20784222483634948,
"loss_toxic": 0.11578925624489785
},
{
"loss": 2.795570468902588,
"loss_size": 0.8643407046794891,
"loss_pdi": 0.4726646453142166,
"loss_ee": 0.6801535218954087,
"loss_delivery": 0.5335889801383018,
"loss_biodist": 0.1555834487080574,
"loss_toxic": 0.11374598890542983
},
{
"loss": 2.5342281222343446,
"loss_size": 0.5860652431845665,
"loss_pdi": 0.47174408435821535,
"loss_ee": 0.6335305064916611,
"loss_delivery": 0.5778901547193527,
"loss_biodist": 0.17659545466303825,
"loss_toxic": 0.11261568909976631
},
{
"loss": 2.269632077217102,
"loss_size": 0.5636066541075706,
"loss_pdi": 0.424516361951828,
"loss_ee": 0.5933005303144455,
"loss_delivery": 0.48853297531604767,
"loss_biodist": 0.15511236637830733,
"loss_toxic": 0.06729794347193092
},
{
"loss": 2.2424973964691164,
"loss_size": 0.48991678208112716,
"loss_pdi": 0.41918782889842987,
"loss_ee": 0.5995140910148621,
"loss_delivery": 0.5058243274688721,
"loss_biodist": 0.15847449898719787,
"loss_toxic": 0.09282479146495462
},
{
"loss": 2.199127399921417,
"loss_size": 0.5160114899277687,
"loss_pdi": 0.42284954339265823,
"loss_ee": 0.5853508979082107,
"loss_delivery": 0.46801379323005676,
"loss_biodist": 0.13517325967550278,
"loss_toxic": 0.09482893161475658
},
{
"loss": 2.1242311239242553,
"loss_size": 0.4818882055580616,
"loss_pdi": 0.4088504880666733,
"loss_ee": 0.5591678321361542,
"loss_delivery": 0.4819866746664047,
"loss_biodist": 0.13510541543364524,
"loss_toxic": 0.08029994852840901
},
{
"loss": 2.1626937627792358,
"loss_size": 0.44518610239028933,
"loss_pdi": 0.4440678536891937,
"loss_ee": 0.6238216310739517,
"loss_delivery": 0.4872687622904778,
"loss_biodist": 0.12052656635642052,
"loss_toxic": 0.07011261922307313
},
{
"loss": 2.1583227634429933,
"loss_size": 0.458985710144043,
"loss_pdi": 0.4431564390659332,
"loss_ee": 0.5801605850458145,
"loss_delivery": 0.4978886917233467,
"loss_biodist": 0.13453557789325715,
"loss_toxic": 0.07013262812979519
}
],
"val": [
{
"loss": 100.50221252441406,
"loss_size": 66.62825012207031,
"loss_pdi": 1.2251655383734033,
"loss_ee": 3.8464752088621026,
"loss_delivery": 23.450343132019043,
"loss_biodist": 5.357328653335571,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.30303030303030304,
"acc_toxic": 1.0
},
{
"loss": 58.95412635803223,
"loss_size": 18.43479824066162,
"loss_pdi": 1.4103724360466003,
"loss_ee": 3.12151378123599,
"loss_delivery": 28.822388648986816,
"loss_biodist": 7.6226441860198975,
"acc_pdi": 0.18181818181818182,
"acc_ee": 0.30303030303030304,
"acc_toxic": 1.0
},
{
"loss": 33.98628044128418,
"loss_size": 6.179022312164307,
"loss_pdi": 0.6798114478588104,
"loss_ee": 1.9242080750409514,
"loss_delivery": 20.320488929748535,
"loss_biodist": 5.317763805389404,
"loss_toxic": 2.2103381169813474e-07,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.30303030303030304,
"acc_toxic": 1.0
},
{
"loss": 11.66386079788208,
"loss_size": 0.6742546819150448,
"loss_pdi": 0.6727426946163177,
"loss_ee": 0.9147305395454168,
"loss_delivery": 7.631983995437622,
"loss_biodist": 1.9435867071151733,
"loss_toxic": 0.004673255629313644,
"acc_pdi": 0.8484848484848485,
"acc_ee": 0.36363636363636365,
"acc_toxic": 1.0
},
{
"loss": 5.221847176551819,
"loss_size": 0.7526329159736633,
"loss_pdi": 1.3156838715076447,
"loss_ee": 1.8227380514144897,
"loss_delivery": 0.6031330544501543,
"loss_biodist": 0.47611144185066223,
"loss_toxic": 0.32870787382125854,
"acc_pdi": 0.8484848484848485,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.88
},
{
"loss": 5.386855244636536,
"loss_size": 1.0329470485448837,
"loss_pdi": 0.9346787482500076,
"loss_ee": 1.940320461988449,
"loss_delivery": 0.8574207723140717,
"loss_biodist": 0.4539874643087387,
"loss_toxic": 0.25234098732471466,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.5454545454545454,
"acc_toxic": 1.0
},
{
"loss": 4.149784326553345,
"loss_size": 0.4728679955005646,
"loss_pdi": 0.9424559473991394,
"loss_ee": 1.6619389355182648,
"loss_delivery": 0.6922308206558228,
"loss_biodist": 0.3804830014705658,
"loss_toxic": 0.07586565986275673,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.6666666666666666,
"acc_toxic": 1.0
},
{
"loss": 3.1718872785568237,
"loss_size": 0.41959452629089355,
"loss_pdi": 0.5824601203203201,
"loss_ee": 1.1527480483055115,
"loss_delivery": 0.6139523684978485,
"loss_biodist": 0.4226408451795578,
"loss_toxic": 0.027006163261830807,
"acc_pdi": 0.8787878787878788,
"acc_ee": 0.696969696969697,
"acc_toxic": 1.0
},
{
"loss": 4.096922159194946,
"loss_size": 0.43914495408535004,
"loss_pdi": 0.7202246338129044,
"loss_ee": 1.6123313009738922,
"loss_delivery": 1.0866239666938782,
"loss_biodist": 0.3203682154417038,
"loss_toxic": 0.007449701661244035,
"acc_pdi": 0.9090909090909091,
"acc_ee": 0.6363636363636364,
"acc_toxic": 1.0
},
{
"loss": 3.72601854801178,
"loss_size": 0.585116520524025,
"loss_pdi": 0.8295658975839615,
"loss_ee": 1.5889590978622437,
"loss_delivery": 0.6002700589597225,
"loss_biodist": 0.19914548844099045,
"loss_toxic": 0.006798723246902227,
"acc_pdi": 0.8787878787878788,
"acc_ee": 0.6666666666666666,
"acc_toxic": 1.0
},
{
"loss": 4.192916035652161,
"loss_size": 0.3904218524694443,
"loss_pdi": 1.1303411722183228,
"loss_ee": 1.88364177942276,
"loss_delivery": 0.5770084410905838,
"loss_biodist": 0.2772811949253082,
"loss_toxic": 0.043116530403494835,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.6363636363636364,
"acc_toxic": 1.0
},
{
"loss": 4.661512494087219,
"loss_size": 0.5379686877131462,
"loss_pdi": 1.0926496386528015,
"loss_ee": 2.1590766310691833,
"loss_delivery": 0.5635539889335632,
"loss_biodist": 0.29881955683231354,
"loss_toxic": 0.13033308740705252,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.6363636363636364,
"acc_toxic": 1.0
},
{
"loss": 4.198178172111511,
"loss_size": 0.27664893865585327,
"loss_pdi": 0.9359037578105927,
"loss_ee": 2.191580146551132,
"loss_delivery": 0.6323938816785812,
"loss_biodist": 0.2183876484632492,
"loss_toxic": 0.06600349466316402,
"acc_pdi": 0.8484848484848485,
"acc_ee": 0.6363636363636364,
"acc_toxic": 1.0
},
{
"loss": 3.994016647338867,
"loss_size": 0.23736193776130676,
"loss_pdi": 0.9927312284708023,
"loss_ee": 2.0256634950637817,
"loss_delivery": 0.5872186869382858,
"loss_biodist": 0.20577140897512436,
"loss_toxic": 0.061211588559672236,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 1.0
},
{
"loss": 4.795967102050781,
"loss_size": 0.4128781408071518,
"loss_pdi": 0.9694665968418121,
"loss_ee": 2.3152187168598175,
"loss_delivery": 0.7095097750425339,
"loss_biodist": 0.28110648691654205,
"loss_toxic": 0.23349096369929612,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 1.0
},
{
"loss": 5.01005482673645,
"loss_size": 0.4783218652009964,
"loss_pdi": 0.9842660576105118,
"loss_ee": 2.408166617155075,
"loss_delivery": 0.6391456425189972,
"loss_biodist": 0.3075519800186157,
"loss_toxic": 0.3175159574020654,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 1.0
},
{
"loss": 5.208752155303955,
"loss_size": 0.4873187690973282,
"loss_pdi": 1.0461187660694122,
"loss_ee": 2.443744570016861,
"loss_delivery": 0.5793078690767288,
"loss_biodist": 0.35986773669719696,
"loss_toxic": 0.41360501595772803,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.96
},
{
"loss": 5.19378924369812,
"loss_size": 0.4627484828233719,
"loss_pdi": 1.058391034603119,
"loss_ee": 2.4470402002334595,
"loss_delivery": 0.5903819352388382,
"loss_biodist": 0.3596591353416443,
"loss_toxic": 0.39829956251196563,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.96
}
]
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_3/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 4, 5, 7, 8, 9, 10, 11, 12, 14, 15, 16, 17, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 32, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 45, 46, 47, 51, 52, 53, 54, 55, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 73, 74, 75, 76, 77, 78, 79, 81, 82, 83, 84, 85, 86, 87, 89, 90, 91, 92, 93, 95, 96, 97, 98, 99, 100, 102, 104, 105, 106, 109, 110, 111, 112, 113, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 127, 129, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 146, 147, 148, 150, 151, 152, 153, 154, 155, 156, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 189, 190, 191, 192, 193, 194, 195, 198, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 212, 213, 214, 215, 216, 217, 218, 219, 221, 223, 224, 226, 227, 228, 230, 231, 233, 234, 235, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 251, 252, 253, 254, 255, 256, 257, 258, 259, 262, 263, 264, 265, 266, 267, 268, 269, 270, 272, 273, 274, 275, 276, 277, 278, 279, 280, 282, 284, 286, 288, 289, 290, 291, 292, 293, 294, 295, 297, 298, 299, 301, 302, 304, 305, 306, 307, 308, 309, 310, 312, 314, 315, 316, 317, 319, 320, 321, 325, 326, 327, 328, 331, 332, 334, 336, 340, 341, 343, 344, 346, 347, 348, 350, 352, 353, 357, 359, 360, 361, 362, 363, 364, 366, 367, 368, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 382, 383, 384, 385, 386, 387, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 400, 401, 402, 403, 404, 406, 407, 410, 411, 412, 414, 415, 416, 418, 419], "outer_test_idx": [3, 6, 13, 18, 19, 26, 33, 44, 48, 49, 50, 56, 57, 72, 80, 88, 94, 101, 103, 107, 108, 114, 126, 128, 130, 145, 149, 157, 167, 171, 175, 188, 196, 197, 199, 211, 220, 222, 225, 229, 232, 236, 250, 260, 261, 271, 281, 283, 285, 287, 296, 300, 303, 311, 313, 318, 322, 323, 324, 329, 330, 333, 335, 337, 338, 339, 342, 345, 349, 351, 354, 355, 356, 358, 365, 369, 381, 388, 399, 405, 408, 409, 413, 417]}

42
models/abl/baseline/20260623_205158/repeat_2/outer_fold_3/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 83,
"mse": 1.7929979497607713,
"rmse": 1.3390287337323166,
"mae": 0.5495479974351509,
"r2": -0.014133306444520155
},
"delivery": {
"n_samples": 59,
"mse": 0.8802460472435503,
"rmse": 0.9382142864205119,
"mae": 0.5970429967362749,
"r2": 0.14429625924896194
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7619047619047619,
"precision": 0.6920821114369502,
"recall": 0.6920821114369502,
"f1": 0.6920821114369502
},
"ee": {
"n_samples": 84,
"accuracy": 0.6309523809523809,
"precision": 0.5623231204626554,
"recall": 0.5826043237807944,
"f1": 0.5665878644602048
},
"toxic": {
"n_samples": 60,
"accuracy": 0.9333333333333333,
"precision": 0.7142857142857143,
"recall": 0.9649122807017544,
"f1": 0.7818181818181817
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.2555339055765927,
"js_divergence": 0.059082110658978844
}
}

12
models/abl/baseline/20260623_205158/repeat_2/outer_fold_4/best_params.json Normal file
View File

@ -0,0 +1,12 @@
{
"dropout": 0.313123672812509,
"lr": 0.0009765192272158632,
"weight_decay": 0.0053740240978641905,
"backbone_lr_ratio": 0.13566938254683006,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_4/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 13}

279
models/abl/baseline/20260623_205158/repeat_2/outer_fold_4/history.json Normal file
View File

@ -0,0 +1,279 @@
{
"train": [
{
"loss": 5.589263248443603,
"loss_size": 1.0933537736535073,
"loss_pdi": 0.6849168121814728,
"loss_ee": 1.0639833092689515,
"loss_delivery": 1.0415786638855935,
"loss_biodist": 1.0585941255092621,
"loss_toxic": 0.6531510829925538
},
{
"loss": 4.902995657920838,
"loss_size": 1.1089929252862931,
"loss_pdi": 0.6317765295505524,
"loss_ee": 0.9846947193145752,
"loss_delivery": 0.8826306372880935,
"loss_biodist": 0.7670059502124786,
"loss_toxic": 0.5441026449203491
},
{
"loss": 4.458215427398682,
"loss_size": 1.009517540037632,
"loss_pdi": 0.5919874012470245,
"loss_ee": 0.9561578929424286,
"loss_delivery": 0.8275555163621903,
"loss_biodist": 0.6069534718990326,
"loss_toxic": 0.48622452914714814
},
{
"loss": 4.075338745117188,
"loss_size": 0.9490635007619858,
"loss_pdi": 0.5800581395626068,
"loss_ee": 0.8821850121021271,
"loss_delivery": 0.8363795816898346,
"loss_biodist": 0.5223618239164353,
"loss_toxic": 0.3257616929709911
},
{
"loss": 3.8364117622375487,
"loss_size": 1.007724793255329,
"loss_pdi": 0.5693021863698959,
"loss_ee": 0.8775955557823181,
"loss_delivery": 0.7496302753686905,
"loss_biodist": 0.42426647543907164,
"loss_toxic": 0.2274421736598015
},
{
"loss": 3.545084571838379,
"loss_size": 0.8552797973155976,
"loss_pdi": 0.5419306427240371,
"loss_ee": 0.8184691488742828,
"loss_delivery": 0.7893632560968399,
"loss_biodist": 0.3621888399124146,
"loss_toxic": 0.19501317739486695
},
{
"loss": 3.3583466053009032,
"loss_size": 0.7890145480632782,
"loss_pdi": 0.5127318739891052,
"loss_ee": 0.8064181625843048,
"loss_delivery": 0.7108009040355683,
"loss_biodist": 0.36283227652311323,
"loss_toxic": 0.19273319020867347
},
{
"loss": 3.3849804639816283,
"loss_size": 0.9505542010068894,
"loss_pdi": 0.5281543344259262,
"loss_ee": 0.7600439131259918,
"loss_delivery": 0.6551562562584877,
"loss_biodist": 0.3498035579919815,
"loss_toxic": 0.15984271727502347
},
{
"loss": 3.154731798171997,
"loss_size": 0.7705727502703666,
"loss_pdi": 0.5095865726470947,
"loss_ee": 0.7537644028663635,
"loss_delivery": 0.6574253737926483,
"loss_biodist": 0.32586769461631776,
"loss_toxic": 0.1539573274552822
},
{
"loss": 3.075357699394226,
"loss_size": 0.7516081839799881,
"loss_pdi": 0.491996830701828,
"loss_ee": 0.7399269461631774,
"loss_delivery": 0.6648136585950851,
"loss_biodist": 0.3128721877932549,
"loss_toxic": 0.13064867593348026
},
{
"loss": 2.982246124744415,
"loss_size": 0.7288039043545723,
"loss_pdi": 0.5127245545387268,
"loss_ee": 0.7571320295333862,
"loss_delivery": 0.585450790822506,
"loss_biodist": 0.30109327137470243,
"loss_toxic": 0.1136342816054821
},
{
"loss": 3.0044891119003294,
"loss_size": 0.7802308201789856,
"loss_pdi": 0.4834390193223953,
"loss_ee": 0.7633242607116699,
"loss_delivery": 0.5924879342317582,
"loss_biodist": 0.28814142793416975,
"loss_toxic": 0.11409494001418352
},
{
"loss": 3.095493721961975,
"loss_size": 0.8713566765189171,
"loss_pdi": 0.483343768119812,
"loss_ee": 0.7427505254745483,
"loss_delivery": 0.6080458179116249,
"loss_biodist": 0.3000340014696121,
"loss_toxic": 0.10869793444871903
}
],
"val": [
{
"loss": 95.50949478149414,
"loss_size": 0.33718644455075264,
"loss_pdi": 6.372642993927002,
"loss_ee": 19.816378593444824,
"loss_delivery": 38.73396682739258,
"loss_biodist": 24.550731658935547,
"loss_toxic": 6.329790115356445,
"acc_pdi": 0.30303030303030304,
"acc_ee": 0.3333333333333333,
"acc_toxic": 0.9090909090909091
},
{
"loss": 62.369991302490234,
"loss_size": 0.23220443353056908,
"loss_pdi": 4.349558711051941,
"loss_ee": 13.490657806396484,
"loss_delivery": 22.331296920776367,
"loss_biodist": 17.440507411956787,
"loss_toxic": 5.3674460649490285,
"acc_pdi": 0.30303030303030304,
"acc_ee": 0.3333333333333333,
"acc_toxic": 0.9090909090909091
},
{
"loss": 31.168947219848633,
"loss_size": 0.345493420958519,
"loss_pdi": 1.7553772032260895,
"loss_ee": 5.731591463088989,
"loss_delivery": 12.164735317230225,
"loss_biodist": 8.648269653320312,
"loss_toxic": 3.0633491800981574,
"acc_pdi": 0.30303030303030304,
"acc_ee": 0.3333333333333333,
"acc_toxic": 0.9090909090909091
},
{
"loss": 5.255474805831909,
"loss_size": 0.20443950593471527,
"loss_pdi": 0.562765508890152,
"loss_ee": 1.0446982681751251,
"loss_delivery": 2.1830610632896423,
"loss_biodist": 0.9487631916999817,
"loss_toxic": 0.363807737827301,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.9090909090909091
},
{
"loss": 1.9932817816734314,
"loss_size": 0.214941649697721,
"loss_pdi": 0.3980669118463993,
"loss_ee": 0.5610466077923775,
"loss_delivery": 0.5538053307682276,
"loss_biodist": 0.18667594343423843,
"loss_toxic": 0.06259655021131039,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.5757575757575758,
"acc_toxic": 1.0
},
{
"loss": 1.9023468494415283,
"loss_size": 0.23417364060878754,
"loss_pdi": 0.3745533050969243,
"loss_ee": 0.583161573857069,
"loss_delivery": 0.44251105515286326,
"loss_biodist": 0.21080515161156654,
"loss_toxic": 0.0419543432071805,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6060606060606061,
"acc_toxic": 1.0
},
{
"loss": 1.9897290766239166,
"loss_size": 0.25809832476079464,
"loss_pdi": 0.371161219663918,
"loss_ee": 0.6089582666754723,
"loss_delivery": 0.4575627539306879,
"loss_biodist": 0.22825980186462402,
"loss_toxic": 0.05499167088419199,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6060606060606061,
"acc_toxic": 1.0
},
{
"loss": 2.0729456543922424,
"loss_size": 0.2479182267561555,
"loss_pdi": 0.3678403962403536,
"loss_ee": 0.6170919463038445,
"loss_delivery": 0.4824555655941367,
"loss_biodist": 0.24085384979844093,
"loss_toxic": 0.10894731059670448,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.6060606060606061,
"acc_toxic": 1.0
},
{
"loss": 2.0146703869104385,
"loss_size": 0.26430252846330404,
"loss_pdi": 0.36510465294122696,
"loss_ee": 0.6197702493518591,
"loss_delivery": 0.47799532441422343,
"loss_biodist": 0.22856901213526726,
"loss_toxic": 0.05289743887260556,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.5757575757575758,
"acc_toxic": 1.0
},
{
"loss": 2.0230285078287125,
"loss_size": 0.265810064971447,
"loss_pdi": 0.3632755377329886,
"loss_ee": 0.6233852598816156,
"loss_delivery": 0.505474029108882,
"loss_biodist": 0.2206256426870823,
"loss_toxic": 0.04034391650930047,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.5757575757575758,
"acc_toxic": 1.0
},
{
"loss": 2.0127016603946686,
"loss_size": 0.2628988870419562,
"loss_pdi": 0.3655702108517289,
"loss_ee": 0.616434583440423,
"loss_delivery": 0.5006475010886788,
"loss_biodist": 0.21789765357971191,
"loss_toxic": 0.04540479928255081,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.5757575757575758,
"acc_toxic": 1.0
},
{
"loss": 2.01269768178463,
"loss_size": 0.26124421413987875,
"loss_pdi": 0.3671441711485386,
"loss_ee": 0.6184194311499596,
"loss_delivery": 0.5057218493893743,
"loss_biodist": 0.2137049399316311,
"loss_toxic": 0.04316108301281929,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6060606060606061,
"acc_toxic": 1.0
},
{
"loss": 2.0035277903079987,
"loss_size": 0.25863443594425917,
"loss_pdi": 0.3646130678243935,
"loss_ee": 0.6178166922181845,
"loss_delivery": 0.5028379475697875,
"loss_biodist": 0.21110595390200615,
"loss_toxic": 0.04502150043845177,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.5757575757575758,
"acc_toxic": 1.0
}
]
}

1
models/abl/baseline/20260623_205158/repeat_2/outer_fold_4/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [1, 2, 3, 4, 6, 7, 8, 9, 10, 12, 13, 14, 15, 16, 17, 18, 19, 20, 22, 23, 25, 26, 28, 29, 31, 33, 34, 35, 37, 41, 42, 43, 44, 45, 46, 48, 49, 50, 51, 52, 53, 54, 56, 57, 59, 60, 62, 63, 64, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 77, 78, 79, 80, 81, 82, 83, 84, 85, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 101, 102, 103, 104, 105, 107, 108, 110, 111, 112, 113, 114, 116, 117, 119, 120, 121, 122, 123, 124, 126, 127, 128, 130, 132, 133, 134, 135, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 149, 152, 154, 155, 157, 160, 161, 162, 163, 164, 165, 166, 167, 169, 170, 171, 173, 174, 175, 176, 177, 178, 180, 181, 182, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 203, 205, 206, 207, 208, 209, 211, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 229, 230, 231, 232, 234, 235, 236, 238, 239, 240, 241, 243, 244, 245, 246, 247, 248, 249, 250, 251, 253, 254, 257, 258, 260, 261, 262, 263, 264, 265, 266, 267, 270, 271, 272, 273, 277, 278, 279, 280, 281, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 298, 299, 300, 301, 302, 303, 304, 305, 306, 309, 311, 312, 313, 314, 315, 316, 318, 319, 320, 322, 323, 324, 325, 326, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 341, 342, 343, 345, 346, 347, 349, 350, 351, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 365, 367, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 381, 383, 384, 385, 386, 388, 390, 391, 392, 393, 394, 395, 396, 397, 399, 400, 401, 402, 403, 405, 406, 407, 408, 409, 410, 411, 413, 415, 416, 417, 418, 419], "outer_test_idx": [0, 5, 11, 21, 24, 27, 30, 32, 36, 38, 39, 40, 47, 55, 58, 61, 65, 76, 86, 95, 100, 106, 109, 115, 118, 125, 129, 131, 136, 148, 150, 151, 153, 156, 158, 159, 168, 172, 179, 183, 184, 185, 202, 204, 210, 212, 213, 228, 233, 237, 242, 252, 255, 256, 259, 268, 269, 274, 275, 276, 282, 297, 307, 308, 310, 317, 321, 327, 340, 344, 348, 352, 364, 366, 368, 379, 380, 382, 387, 389, 398, 404, 412, 414]}

42
models/abl/baseline/20260623_205158/repeat_2/outer_fold_4/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.6443346713585372,
"rmse": 0.8027045978182368,
"mae": 0.5012692398380577,
"r2": 0.1378001731045243
},
"delivery": {
"n_samples": 58,
"mse": 1.091911519660062,
"rmse": 1.044945701775964,
"mae": 0.6530626353904091,
"r2": -0.05586041412459375
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7619047619047619,
"precision": 0.711111111111111,
"recall": 0.750733137829912,
"f1": 0.7214854111405835
},
"ee": {
"n_samples": 84,
"accuracy": 0.6785714285714286,
"precision": 0.6048309178743961,
"recall": 0.638034188034188,
"f1": 0.6173894191612906
},
"toxic": {
"n_samples": 59,
"accuracy": 0.9661016949152542,
"precision": 0.8,
"recall": 0.9821428571428572,
"f1": 0.865909090909091
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.2916542945348232,
"js_divergence": 0.08128207465162181
}
}

60
models/abl/baseline/20260623_205158/strata_info.json Normal file
View File

@ -0,0 +1,60 @@
{
"original_strata_counts": {
"T0|P0|E0": "5",
"T0|P0|E1": "58",
"T0|P0|E2": "169",
"T0|P1|E0": "1",
"T0|P1|E1": "17",
"T0|P1|E2": "32",
"T0|P2|E2": "3",
"T1|P0|E2": "9",
"T1|P1|E2": "5",
"TNA|P0|E0": "28",
"TNA|P0|E1": "21",
"TNA|P0|E2": "20",
"TNA|P1|E0": "29",
"TNA|P1|E1": "7",
"TNA|P1|E2": "14",
"TNA|P2|E2": "1",
"TNA|P3|E0": "1"
},
"rare_strata": [
"T0|P1|E0",
"T0|P2|E2",
"TNA|P2|E2",
"TNA|P3|E0"
],
"final_strata": [
"RARE",
"T0|P0|E0",
"T0|P0|E1",
"T0|P0|E2",
"T0|P1|E1",
"T0|P1|E2",
"T1|P0|E2",
"T1|P1|E2",
"TNA|P0|E0",
"TNA|P0|E1",
"TNA|P0|E2",
"TNA|P1|E0",
"TNA|P1|E1",
"TNA|P1|E2"
],
"final_strata_counts": {
"RARE": "6",
"T0|P0|E0": "5",
"T0|P0|E1": "58",
"T0|P0|E2": "169",
"T0|P1|E1": "17",
"T0|P1|E2": "32",
"T1|P0|E2": "9",
"T1|P1|E2": "5",
"TNA|P0|E0": "28",
"TNA|P0|E1": "21",
"TNA|P0|E2": "20",
"TNA|P1|E0": "29",
"TNA|P1|E1": "7",
"TNA|P1|E2": "14"
},
"n_rare_merged": "6"
}

932
models/abl/baseline/20260623_205158/summary.json Normal file
View File

@ -0,0 +1,932 @@
{
"fold_results": [
{
"fold": 0,
"best_params": {
"dropout": 0.18569939775458322,
"lr": 0.0009872295492725695,
"weight_decay": 0.017393516831905452,
"backbone_lr_ratio": 0.17239541174804215,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 11,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.5282298486963632,
"rmse": 1.2362159393473144,
"mae": 0.5704747740717897,
"r2": 0.06468571536968037
},
"delivery": {
"n_samples": 58,
"mse": 0.5436139978785772,
"rmse": 0.7373018363455887,
"mae": 0.5591465608570082,
"r2": 0.3090338048982032
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7738095238095238,
"precision": 0.7232142857142857,
"recall": 0.7494654312188168,
"f1": 0.7325289089994973
},
"ee": {
"n_samples": 84,
"accuracy": 0.7023809523809523,
"precision": 0.6590590922487474,
"recall": 0.6976190476190475,
"f1": 0.666253917297256
},
"toxic": {
"n_samples": 58,
"accuracy": 0.9655172413793104,
"precision": 0.75,
"recall": 0.9821428571428572,
"f1": 0.8242424242424242
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.33174143107041487,
"js_divergence": 0.07276002005717812
}
}
},
{
"fold": 1,
"best_params": {
"dropout": 0.36640960296080094,
"lr": 0.0009935032389272061,
"weight_decay": 0.007506834667713186,
"backbone_lr_ratio": 0.10108281143505408,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 11,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.37057228076561216,
"rmse": 0.6087464831648821,
"mae": 0.4609523837321571,
"r2": -0.13159599860936777
},
"delivery": {
"n_samples": 61,
"mse": 1.082995768972751,
"rmse": 1.0406708264253164,
"mae": 0.6707069295779114,
"r2": 0.18025027489571332
},
"pdi": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.5886939571150097,
"recall": 0.6031746031746031,
"f1": 0.5916666666666667
},
"ee": {
"n_samples": 84,
"accuracy": 0.5595238095238095,
"precision": 0.4957294957294957,
"recall": 0.49526663812378097,
"f1": 0.4845300550726907
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9508196721311475,
"precision": 0.75,
"recall": 0.9741379310344828,
"f1": 0.8200589970501475
},
"biodist": {
"n_samples": 61,
"kl_divergence": 0.37009493729038634,
"js_divergence": 0.09345595384422643
}
}
},
{
"fold": 2,
"best_params": {
"dropout": 0.318342405407684,
"lr": 0.0008937362022799393,
"weight_decay": 0.06633867694372529,
"backbone_lr_ratio": 0.20906844321970927,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 10,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.365749434655222,
"rmse": 0.6047722171654565,
"mae": 0.47718314868619754,
"r2": 0.2776514013890329
},
"delivery": {
"n_samples": 60,
"mse": 0.668501323406245,
"rmse": 0.817619302246617,
"mae": 0.5434193710253263,
"r2": 0.11010887128905211
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7738095238095238,
"precision": 0.7122033898305085,
"recall": 0.7294721407624634,
"f1": 0.719359943731317
},
"ee": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.6145962732919253,
"recall": 0.6785714285714285,
"f1": 0.6278721278721279
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9344262295081968,
"precision": 0.7142857142857143,
"recall": 0.9655172413793103,
"f1": 0.7821428571428571
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.3336122920652159,
"js_divergence": 0.08430196229187475
}
}
},
{
"fold": 3,
"best_params": {
"dropout": 0.12612237693012526,
"lr": 0.0009464809649969671,
"weight_decay": 0.07051438027639334,
"backbone_lr_ratio": 0.9343703462702573,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 18,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.7046487347114636,
"rmse": 1.3056219723608604,
"mae": 0.5513793248122176,
"r2": 0.03583768297668177
},
"delivery": {
"n_samples": 59,
"mse": 0.7345580643934568,
"rmse": 0.8570636291393171,
"mae": 0.5434487631733892,
"r2": 0.28592228790047713
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7380952380952381,
"precision": 0.6612903225806452,
"recall": 0.6612903225806452,
"f1": 0.6612903225806452
},
"ee": {
"n_samples": 84,
"accuracy": 0.6547619047619048,
"precision": 0.6206349206349207,
"recall": 0.5979889391654097,
"f1": 0.5781687846203974
},
"toxic": {
"n_samples": 60,
"accuracy": 0.9333333333333333,
"precision": 0.7142857142857143,
"recall": 0.9649122807017544,
"f1": 0.7818181818181817
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.2397795651876451,
"js_divergence": 0.0585131523487816
}
}
},
{
"fold": 4,
"best_params": {
"dropout": 0.313123672812509,
"lr": 0.0009765192272158632,
"weight_decay": 0.0053740240978641905,
"backbone_lr_ratio": 0.13566938254683006,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 13,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.6868306794055117,
"rmse": 0.8287524838005083,
"mae": 0.5001247360376597,
"r2": 0.08093523565734784
},
"delivery": {
"n_samples": 58,
"mse": 1.1008487633192388,
"rmse": 1.0492134021824344,
"mae": 0.654239107951008,
"r2": -0.06450258120608798
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6752941176470588,
"recall": 0.718475073313783,
"f1": 0.6785714285714286
},
"ee": {
"n_samples": 84,
"accuracy": 0.6785714285714286,
"precision": 0.626984126984127,
"recall": 0.6594017094017094,
"f1": 0.6342062193126022
},
"toxic": {
"n_samples": 59,
"accuracy": 0.9661016949152542,
"precision": 0.8,
"recall": 0.9821428571428572,
"f1": 0.865909090909091
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.2992118497501824,
"js_divergence": 0.08211837700207104
}
}
},
{
"fold": 0,
"best_params": {
"dropout": 0.18569939775458322,
"lr": 0.0009872295492725695,
"weight_decay": 0.017393516831905452,
"backbone_lr_ratio": 0.17239541174804215,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 11,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.5000130580754827,
"rmse": 1.2247502023169796,
"mae": 0.5513235701315374,
"r2": 0.08195508578319866
},
"delivery": {
"n_samples": 58,
"mse": 0.6098699437647568,
"rmse": 0.7809417031794094,
"mae": 0.6011813956088033,
"r2": 0.2248184995335506
},
"pdi": {
"n_samples": 84,
"accuracy": 0.8333333333333334,
"precision": 0.7904490377761939,
"recall": 0.7904490377761939,
"f1": 0.7904490377761939
},
"ee": {
"n_samples": 84,
"accuracy": 0.6904761904761905,
"precision": 0.6475648393751842,
"recall": 0.6809523809523809,
"f1": 0.6526484751203853
},
"toxic": {
"n_samples": 58,
"accuracy": 0.9827586206896551,
"precision": 0.9912280701754386,
"recall": 0.75,
"f1": 0.8289085545722714
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.31929652154466054,
"js_divergence": 0.06311757696370286
}
}
},
{
"fold": 1,
"best_params": {
"dropout": 0.36640960296080094,
"lr": 0.0009935032389272061,
"weight_decay": 0.007506834667713186,
"backbone_lr_ratio": 0.10108281143505408,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 11,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.35953046688813006,
"rmse": 0.5996085947417116,
"mae": 0.4586696324404329,
"r2": -0.09787822464275231
},
"delivery": {
"n_samples": 61,
"mse": 1.1161931438403756,
"rmse": 1.0565004230194968,
"mae": 0.6922882909840736,
"r2": 0.15512225528421297
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7261904761904762,
"precision": 0.6685185185185185,
"recall": 0.7063492063492063,
"f1": 0.6762192056309704
},
"ee": {
"n_samples": 84,
"accuracy": 0.5476190476190477,
"precision": 0.48175125313283207,
"recall": 0.48011512297226583,
"f1": 0.4721157939548744
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9508196721311475,
"precision": 0.75,
"recall": 0.9741379310344828,
"f1": 0.8200589970501475
},
"biodist": {
"n_samples": 61,
"kl_divergence": 0.3950644341107201,
"js_divergence": 0.09966091530608283
}
}
},
{
"fold": 2,
"best_params": {
"dropout": 0.318342405407684,
"lr": 0.0008937362022799393,
"weight_decay": 0.06633867694372529,
"backbone_lr_ratio": 0.20906844321970927,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 10,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.4926918006241934,
"rmse": 0.7019200813655307,
"mae": 0.537246183603115,
"r2": 0.026942496675384175
},
"delivery": {
"n_samples": 60,
"mse": 0.726897839382552,
"rmse": 0.8525830395818064,
"mae": 0.5536939328614001,
"r2": 0.03237298701263025
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7261904761904762,
"precision": 0.6686520376175549,
"recall": 0.6972140762463344,
"f1": 0.6762192056309704
},
"ee": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.6139316239316239,
"recall": 0.6785714285714285,
"f1": 0.6277576489533011
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9344262295081968,
"precision": 0.7142857142857143,
"recall": 0.9655172413793103,
"f1": 0.7821428571428571
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.38696828264806826,
"js_divergence": 0.0987549780391864
}
}
},
{
"fold": 3,
"best_params": {
"dropout": 0.12612237693012526,
"lr": 0.0009464809649969671,
"weight_decay": 0.07051438027639334,
"backbone_lr_ratio": 0.9343703462702573,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 18,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.7722816318702095,
"rmse": 1.3312706831708605,
"mae": 0.5694137913208589,
"r2": -0.0024159991478140963
},
"delivery": {
"n_samples": 59,
"mse": 0.6715869625788514,
"rmse": 0.8195040955229275,
"mae": 0.48721023672654973,
"r2": 0.34713768051792715
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7023809523809523,
"precision": 0.6267796610169492,
"recall": 0.6370967741935484,
"f1": 0.6307367680675224
},
"ee": {
"n_samples": 84,
"accuracy": 0.6190476190476191,
"precision": 0.5535293250810492,
"recall": 0.5760683760683761,
"f1": 0.5530481924570595
},
"toxic": {
"n_samples": 60,
"accuracy": 0.9333333333333333,
"precision": 0.7142857142857143,
"recall": 0.9649122807017544,
"f1": 0.7818181818181817
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.27026190922231863,
"js_divergence": 0.06299181317830951
}
}
},
{
"fold": 4,
"best_params": {
"dropout": 0.313123672812509,
"lr": 0.0009765192272158632,
"weight_decay": 0.0053740240978641905,
"backbone_lr_ratio": 0.13566938254683006,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 13,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.7536341327012797,
"rmse": 0.8681210357440255,
"mae": 0.5072707110472644,
"r2": -0.00845608290415334
},
"delivery": {
"n_samples": 58,
"mse": 1.087167994986309,
"rmse": 1.0426734843594656,
"mae": 0.6497257278654082,
"r2": -0.0512735040716632
},
"pdi": {
"n_samples": 84,
"accuracy": 0.75,
"precision": 0.6890838206627681,
"recall": 0.7133431085043989,
"f1": 0.6974789915966386
},
"ee": {
"n_samples": 84,
"accuracy": 0.6904761904761905,
"precision": 0.6308992939427722,
"recall": 0.6444444444444445,
"f1": 0.633663610371685
},
"toxic": {
"n_samples": 59,
"accuracy": 0.9661016949152542,
"precision": 0.8,
"recall": 0.9821428571428572,
"f1": 0.865909090909091
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.21472605299515538,
"js_divergence": 0.0586261381599948
}
}
},
{
"fold": 0,
"best_params": {
"dropout": 0.18569939775458322,
"lr": 0.0009872295492725695,
"weight_decay": 0.017393516831905452,
"backbone_lr_ratio": 0.17239541174804215,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 11,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.5072429789639405,
"rmse": 1.2276982442619768,
"mae": 0.5558247003165713,
"r2": 0.07753019623566748
},
"delivery": {
"n_samples": 58,
"mse": 0.5835939668070607,
"rmse": 0.76393322143173,
"mae": 0.5947980444942569,
"r2": 0.2582168518421626
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7738095238095238,
"precision": 0.7232142857142857,
"recall": 0.7494654312188168,
"f1": 0.7325289089994973
},
"ee": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6729142936039488,
"recall": 0.7214285714285715,
"f1": 0.6806141997028696
},
"toxic": {
"n_samples": 58,
"accuracy": 0.9655172413793104,
"precision": 0.75,
"recall": 0.9821428571428572,
"f1": 0.8242424242424242
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.28982565124202386,
"js_divergence": 0.06069208833134297
}
}
},
{
"fold": 1,
"best_params": {
"dropout": 0.36640960296080094,
"lr": 0.0009935032389272061,
"weight_decay": 0.007506834667713186,
"backbone_lr_ratio": 0.10108281143505408,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 11,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.4057494802572597,
"rmse": 0.6369846781966264,
"mae": 0.5019206670778138,
"r2": -0.2390146595647713
},
"delivery": {
"n_samples": 61,
"mse": 1.095192634861098,
"rmse": 1.0465145172720243,
"mae": 0.6369023019234177,
"r2": 0.17101812667726723
},
"pdi": {
"n_samples": 84,
"accuracy": 0.6190476190476191,
"precision": 0.5575167376749848,
"recall": 0.5714285714285714,
"f1": 0.5543766578249337
},
"ee": {
"n_samples": 84,
"accuracy": 0.5714285714285714,
"precision": 0.5153903903903904,
"recall": 0.5209076637648066,
"f1": 0.507505844715147
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9508196721311475,
"precision": 0.75,
"recall": 0.9741379310344828,
"f1": 0.8200589970501475
},
"biodist": {
"n_samples": 61,
"kl_divergence": 0.3834603232909285,
"js_divergence": 0.0914682060621546
}
}
},
{
"fold": 2,
"best_params": {
"dropout": 0.318342405407684,
"lr": 0.0008937362022799393,
"weight_decay": 0.06633867694372529,
"backbone_lr_ratio": 0.20906844321970927,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 10,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.4667056344892945,
"rmse": 0.6831585719943024,
"mae": 0.5159196099488154,
"r2": 0.07826471049783545
},
"delivery": {
"n_samples": 60,
"mse": 0.6618025560042317,
"rmse": 0.8135124805460772,
"mae": 0.5426560255466029,
"r2": 0.11902609175763024
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6517857142857143,
"recall": 0.6744868035190617,
"f1": 0.6583050847457628
},
"ee": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.6138888888888889,
"recall": 0.6785714285714285,
"f1": 0.6291768416768416
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9344262295081968,
"precision": 0.7142857142857143,
"recall": 0.9655172413793103,
"f1": 0.7821428571428571
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.3327396562359482,
"js_divergence": 0.08480501673906925
}
}
},
{
"fold": 3,
"best_params": {
"dropout": 0.12612237693012526,
"lr": 0.0009464809649969671,
"weight_decay": 0.07051438027639334,
"backbone_lr_ratio": 0.9343703462702573,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 18,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.7929979497607713,
"rmse": 1.3390287337323166,
"mae": 0.5495479974351509,
"r2": -0.014133306444520155
},
"delivery": {
"n_samples": 59,
"mse": 0.8802460472435503,
"rmse": 0.9382142864205119,
"mae": 0.5970429967362749,
"r2": 0.14429625924896194
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7619047619047619,
"precision": 0.6920821114369502,
"recall": 0.6920821114369502,
"f1": 0.6920821114369502
},
"ee": {
"n_samples": 84,
"accuracy": 0.6309523809523809,
"precision": 0.5623231204626554,
"recall": 0.5826043237807944,
"f1": 0.5665878644602048
},
"toxic": {
"n_samples": 60,
"accuracy": 0.9333333333333333,
"precision": 0.7142857142857143,
"recall": 0.9649122807017544,
"f1": 0.7818181818181817
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.2555339055765927,
"js_divergence": 0.059082110658978844
}
}
},
{
"fold": 4,
"best_params": {
"dropout": 0.313123672812509,
"lr": 0.0009765192272158632,
"weight_decay": 0.0053740240978641905,
"backbone_lr_ratio": 0.13566938254683006,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
},
"epoch_mean": 13,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.6443346713585372,
"rmse": 0.8027045978182368,
"mae": 0.5012692398380577,
"r2": 0.1378001731045243
},
"delivery": {
"n_samples": 58,
"mse": 1.091911519660062,
"rmse": 1.044945701775964,
"mae": 0.6530626353904091,
"r2": -0.05586041412459375
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7619047619047619,
"precision": 0.711111111111111,
"recall": 0.750733137829912,
"f1": 0.7214854111405835
},
"ee": {
"n_samples": 84,
"accuracy": 0.6785714285714286,
"precision": 0.6048309178743961,
"recall": 0.638034188034188,
"f1": 0.6173894191612906
},
"toxic": {
"n_samples": 59,
"accuracy": 0.9661016949152542,
"precision": 0.8,
"recall": 0.9821428571428572,
"f1": 0.865909090909091
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.2916542945348232,
"js_divergence": 0.08128207465162181
}
}
}
],
"summary_stats": {
"size": {
"mse_mean": 0.9567475188815514,
"mse_std": 0.569730318457903,
"rmse_mean": 0.9332903012787724,
"rmse_std": 0.2927742004011441,
"mae_mean": 0.5205680313666426,
"mae_std": 0.036274021744309826,
"r2_mean": 0.024540561758398263,
"r2_std": 0.11631746182529404
},
"delivery": {
"mse_mean": 0.8436653684732743,
"mse_std": 0.2182433415975543,
"rmse_mean": 0.9107461299632457,
"rmse_std": 0.11919251331457457,
"mae_mean": 0.598634821381456,
"mae_std": 0.05735289715081742,
"r2_mean": 0.1443791660970296,
"r2_std": 0.1281061991656446
},
"pdi": {
"accuracy_mean": 0.7357142857142857,
"accuracy_std": 0.049026810195176206,
"precision_mean": 0.6759926072468357,
"precision_std": 0.05509185976342328,
"recall_mean": 0.6963017219702202,
"recall_std": 0.057127965795130256,
"f1_mean": 0.6808865768933052,
"f1_std": 0.056739009627821094
},
"ee": {
"accuracy_mean": 0.6492063492063491,
"accuracy_std": 0.05085561696381203,
"precision_mean": 0.5942685237048638,
"precision_std": 0.05709593570250263,
"recall_mean": 0.6220363794313374,
"recall_std": 0.07370705460979522,
"f1_mean": 0.5954359329832487,
"f1_std": 0.0634519507679728
},
"toxic": {
"accuracy_mean": 0.9511890595408049,
"accuracy_std": 0.016071213375949578,
"precision_mean": 0.7617961570593149,
"precision_std": 0.0689235340767993,
"recall_mean": 0.958294443004062,
"recall_std": 0.05613748087593979,
"f1_mean": 0.8151453855878633,
"f1_std": 0.03136647604932657
},
"biodist": {
"kl_divergence_mean": 0.31426474045100566,
"kl_divergence_std": 0.05350543992895652,
"js_divergence_mean": 0.07677535890897173,
"js_divergence_std": 0.01482841183016674
}
}
}

16
models/abl/both_biot5/20260624_015624/outer_fold_0/best_params.json Normal file
View File

@ -0,0 +1,16 @@
{
"dropout": 0.38118503457897024,
"lr": 0.0009770749364761302,
"weight_decay": 0.0008416934971427061,
"backbone_lr_ratio": 0.048484014956722714,
"moe_n_experts": 2,
"moe_top_k": 2,
"moe_expert_hidden_mult": 1,
"llm_lora_r": 16,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/both_biot5/20260624_015624/outer_fold_0/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 15}

351
models/abl/both_biot5/20260624_015624/outer_fold_0/history.json Normal file
View File

@ -0,0 +1,351 @@
{
"train": [
{
"loss": 5.629639577865601,
"loss_size": 0.8370314225554466,
"loss_pdi": 0.7166214644908905,
"loss_ee": 1.1042668342590332,
"loss_delivery": 1.1503031998872757,
"loss_biodist": 1.1618747770786286,
"loss_toxic": 0.6466398298740387,
"loss_moe_lb": 2.0
},
{
"loss": 5.174609231948852,
"loss_size": 1.0997996091842652,
"loss_pdi": 0.6605279207229614,
"loss_ee": 1.0257868766784668,
"loss_delivery": 0.9808838427066803,
"loss_biodist": 0.8544291496276856,
"loss_toxic": 0.551453810930252,
"loss_moe_lb": 1.999999988079071
},
{
"loss": 4.431454110145569,
"loss_size": 0.7972716838121414,
"loss_pdi": 0.6387695252895356,
"loss_ee": 0.9912315964698791,
"loss_delivery": 0.9170992910861969,
"loss_biodist": 0.6875045835971832,
"loss_toxic": 0.4016995593905449,
"loss_moe_lb": 2.000000023841858
},
{
"loss": 3.8852075576782226,
"loss_size": 0.7344662338495255,
"loss_pdi": 0.6063814699649811,
"loss_ee": 0.963562548160553,
"loss_delivery": 0.8228829815983772,
"loss_biodist": 0.49548698365688326,
"loss_toxic": 0.261992035806179,
"loss_moe_lb": 1.999999988079071
},
{
"loss": 3.714037370681763,
"loss_size": 0.7619987577199936,
"loss_pdi": 0.5843621015548706,
"loss_ee": 0.881328821182251,
"loss_delivery": 0.8246007680892944,
"loss_biodist": 0.41834571659564973,
"loss_toxic": 0.24274032153189182,
"loss_moe_lb": 2.000000011920929
},
{
"loss": 3.563366413116455,
"loss_size": 0.7258953899145126,
"loss_pdi": 0.5627626419067383,
"loss_ee": 0.8682501256465912,
"loss_delivery": 0.8690563052892685,
"loss_biodist": 0.3803571105003357,
"loss_toxic": 0.15887218862771987,
"loss_moe_lb": 1.9999999761581422
},
{
"loss": 3.3824042320251464,
"loss_size": 0.6917768061161041,
"loss_pdi": 0.5605458229780197,
"loss_ee": 0.8651491045951843,
"loss_delivery": 0.8123000666499138,
"loss_biodist": 0.3316554129123688,
"loss_toxic": 0.1240404974669218,
"loss_moe_lb": 2.000000023841858
},
{
"loss": 3.302357888221741,
"loss_size": 0.6924583867192269,
"loss_pdi": 0.5409035980701447,
"loss_ee": 0.846926087141037,
"loss_delivery": 0.815614002943039,
"loss_biodist": 0.2887734815478325,
"loss_toxic": 0.1221643571741879,
"loss_moe_lb": 1.999999988079071
},
{
"loss": 3.2315545797348024,
"loss_size": 0.6983503788709641,
"loss_pdi": 0.5339838176965713,
"loss_ee": 0.8057428479194642,
"loss_delivery": 0.8637148022651673,
"loss_biodist": 0.24896689057350158,
"loss_toxic": 0.08789459783583879,
"loss_moe_lb": 1.999999988079071
},
{
"loss": 3.0863017082214355,
"loss_size": 0.6584484815597534,
"loss_pdi": 0.5230895698070526,
"loss_ee": 0.8105671405792236,
"loss_delivery": 0.7606720671057701,
"loss_biodist": 0.2532289892435074,
"loss_toxic": 0.08702232539653779,
"loss_moe_lb": 2.0
},
{
"loss": 3.3488226175308227,
"loss_size": 0.9569226488471031,
"loss_pdi": 0.5225043386220932,
"loss_ee": 0.8297718048095704,
"loss_delivery": 0.7143090248107911,
"loss_biodist": 0.27392726168036463,
"loss_toxic": 0.06445237565785647,
"loss_moe_lb": 2.000000023841858
},
{
"loss": 2.9060803294181823,
"loss_size": 0.5815626472234726,
"loss_pdi": 0.5093928277492523,
"loss_ee": 0.7667526185512543,
"loss_delivery": 0.747194854915142,
"loss_biodist": 0.23200357258319854,
"loss_toxic": 0.07565911188721657,
"loss_moe_lb": 2.000000023841858
},
{
"loss": 3.0850929975509644,
"loss_size": 0.8070279285311699,
"loss_pdi": 0.48761967420578,
"loss_ee": 0.7690303862094879,
"loss_delivery": 0.7080013066530227,
"loss_biodist": 0.24257410168647767,
"loss_toxic": 0.07937230654060841,
"loss_moe_lb": 2.000000011920929
},
{
"loss": 2.8702262163162233,
"loss_size": 0.5871895246207715,
"loss_pdi": 0.5188780128955841,
"loss_ee": 0.7922944843769073,
"loss_delivery": 0.6827648490667343,
"loss_biodist": 0.19578708112239837,
"loss_toxic": 0.09972089696675539,
"loss_moe_lb": 2.0
},
{
"loss": 2.7740705251693725,
"loss_size": 0.548411799967289,
"loss_pdi": 0.5007277846336364,
"loss_ee": 0.7405028104782104,
"loss_delivery": 0.7051813542842865,
"loss_biodist": 0.2051745280623436,
"loss_toxic": 0.07857431825250387,
"loss_moe_lb": 1.999999988079071
}
],
"val": [
{
"loss": 326.16290283203125,
"loss_size": 120.44477081298828,
"loss_pdi": 5.122471332550049,
"loss_ee": 2.383893821388483,
"loss_delivery": 194.1770782470703,
"loss_biodist": 2.57398784160614,
"loss_toxic": 2.32453191280365,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.7575757575757576,
"acc_toxic": 0.0
},
{
"loss": 106.2081527709961,
"loss_size": 49.1336727142334,
"loss_pdi": 3.6693177223205,
"loss_ee": 2.199763808399439,
"loss_delivery": 48.561180114746094,
"loss_biodist": 3.2662585973739624,
"loss_toxic": 0.12828633561730385,
"loss_moe_lb": 1.9999998211860657,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.7575757575757576,
"acc_toxic": 1.0
},
{
"loss": 29.09860897064209,
"loss_size": 12.30441665649414,
"loss_pdi": 2.401897069717961,
"loss_ee": 1.3710492867976427,
"loss_delivery": 10.599392414093018,
"loss_biodist": 2.321728855371475,
"loss_toxic": 0.41285043209791183,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.7575757575757576,
"acc_toxic": 1.0
},
{
"loss": 7.5232861042022705,
"loss_size": 1.8920176029205322,
"loss_pdi": 1.1291465051472187,
"loss_ee": 0.809601366519928,
"loss_delivery": 1.6841156333684921,
"loss_biodist": 0.8365733176469803,
"loss_toxic": 1.227177157998085,
"loss_moe_lb": 2.0,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.7272727272727273,
"acc_toxic": 0.9
},
{
"loss": 5.164956092834473,
"loss_size": 0.358419019728899,
"loss_pdi": 1.2186900079250336,
"loss_ee": 0.6728924587368965,
"loss_delivery": 0.5422201752662659,
"loss_biodist": 0.21467457339167595,
"loss_toxic": 2.1638595312833786,
"loss_moe_lb": 1.9999998807907104,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.3939393939393939,
"acc_toxic": 0.9
},
{
"loss": 4.1869136095047,
"loss_size": 0.283981516957283,
"loss_pdi": 1.156490445137024,
"loss_ee": 0.6733633950352669,
"loss_delivery": 0.38626348320394754,
"loss_biodist": 0.26169781386852264,
"loss_toxic": 1.4192646592855453,
"loss_moe_lb": 2.0,
"acc_pdi": 0.7878787878787878,
"acc_ee": 0.3939393939393939,
"acc_toxic": 0.9
},
{
"loss": 4.0673065185546875,
"loss_size": 0.2509373798966408,
"loss_pdi": 1.2529352903366089,
"loss_ee": 0.6407689638435841,
"loss_delivery": 0.3957926770672202,
"loss_biodist": 0.2356797456741333,
"loss_toxic": 1.2829149067401886,
"loss_moe_lb": 2.0,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 4.362272620201111,
"loss_size": 0.23665804788470268,
"loss_pdi": 1.4687186479568481,
"loss_ee": 0.5836386568844318,
"loss_delivery": 0.3769393851980567,
"loss_biodist": 0.22988024167716503,
"loss_toxic": 1.45628771930933,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.6363636363636364,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 3.909448027610779,
"loss_size": 0.22452042810618877,
"loss_pdi": 1.4854494333267212,
"loss_ee": 0.5961620267480612,
"loss_delivery": 0.3709214013069868,
"loss_biodist": 0.21133559942245483,
"loss_toxic": 1.0162529349327087,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6363636363636364,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 3.5202027559280396,
"loss_size": 0.2468378245830536,
"loss_pdi": 1.420798122882843,
"loss_ee": 0.5892483126372099,
"loss_delivery": 0.41495488211512566,
"loss_biodist": 0.2065230067819357,
"loss_toxic": 0.6433024294674397,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 3.5848307609558105,
"loss_size": 0.2656059116125107,
"loss_pdi": 1.4705984592437744,
"loss_ee": 0.5635334458202124,
"loss_delivery": 0.4237087741494179,
"loss_biodist": 0.18785163201391697,
"loss_toxic": 0.6749111115932465,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 3.740071415901184,
"loss_size": 0.2570607364177704,
"loss_pdi": 1.5505283176898956,
"loss_ee": 0.5641665607690811,
"loss_delivery": 0.42181940376758575,
"loss_biodist": 0.1926831342279911,
"loss_toxic": 0.755558256059885,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6060606060606061,
"acc_ee": 0.5151515151515151,
"acc_toxic": 0.9
},
{
"loss": 3.755470633506775,
"loss_size": 0.29488252103328705,
"loss_pdi": 1.6161202490329742,
"loss_ee": 0.5678785536438227,
"loss_delivery": 0.4295586571097374,
"loss_biodist": 0.19815191626548767,
"loss_toxic": 0.6567674949765205,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.6060606060606061,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 3.756665587425232,
"loss_size": 0.28851318359375,
"loss_pdi": 1.6128602027893066,
"loss_ee": 0.5694516357034445,
"loss_delivery": 0.4351465404033661,
"loss_biodist": 0.19257093593478203,
"loss_toxic": 0.6656300947070122,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6060606060606061,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
},
{
"loss": 3.9940130710601807,
"loss_size": 0.2515711784362793,
"loss_pdi": 1.6328342258930206,
"loss_ee": 0.5652816658839583,
"loss_delivery": 0.42837221175432205,
"loss_biodist": 0.19714058563113213,
"loss_toxic": 0.9184133037924767,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.6060606060606061,
"acc_ee": 0.48484848484848486,
"acc_toxic": 0.9
}
]
}

1
models/abl/both_biot5/20260624_015624/outer_fold_0/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 21, 22, 23, 24, 25, 26, 27, 28, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 52, 53, 54, 55, 56, 57, 58, 61, 62, 63, 64, 65, 66, 67, 68, 71, 72, 73, 75, 76, 77, 80, 81, 85, 86, 88, 89, 90, 92, 94, 95, 96, 98, 99, 100, 101, 103, 104, 105, 106, 107, 108, 109, 111, 113, 114, 115, 116, 117, 118, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 133, 134, 135, 136, 138, 139, 141, 144, 145, 146, 147, 148, 149, 150, 151, 153, 154, 155, 156, 157, 158, 159, 160, 163, 164, 167, 168, 169, 170, 171, 172, 175, 176, 178, 179, 180, 182, 183, 184, 185, 188, 189, 190, 191, 192, 193, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 206, 208, 210, 211, 212, 213, 216, 217, 218, 219, 220, 222, 224, 225, 227, 228, 229, 230, 231, 232, 233, 234, 236, 237, 239, 241, 242, 244, 245, 247, 248, 249, 250, 252, 253, 255, 256, 257, 258, 259, 260, 261, 262, 265, 267, 268, 269, 270, 271, 272, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 285, 286, 287, 288, 291, 292, 294, 296, 297, 298, 299, 300, 301, 303, 304, 306, 307, 308, 309, 310, 311, 312, 313, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 332, 333, 335, 336, 337, 338, 339, 340, 342, 343, 344, 345, 346, 347, 348, 349, 351, 352, 354, 355, 356, 357, 358, 359, 360, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 375, 377, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 393, 395, 396, 397, 398, 399, 400, 401, 403, 404, 405, 406, 407, 408, 409, 411, 412, 413, 414, 415, 417, 418, 419], "outer_test_idx": [7, 16, 20, 29, 51, 59, 60, 69, 70, 74, 78, 79, 82, 83, 84, 87, 91, 93, 97, 102, 110, 112, 119, 120, 121, 132, 137, 140, 142, 143, 152, 161, 162, 165, 166, 173, 174, 177, 181, 186, 187, 194, 205, 207, 209, 214, 215, 221, 223, 226, 235, 238, 240, 243, 246, 251, 254, 263, 264, 266, 273, 284, 289, 290, 293, 295, 302, 305, 314, 315, 331, 334, 341, 350, 353, 361, 373, 376, 378, 392, 394, 402, 410, 416]}

42
models/abl/both_biot5/20260624_015624/outer_fold_0/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 83,
"mse": 1.2775270638012577,
"rmse": 1.1302774278031291,
"mae": 0.5309888495648482,
"r2": 0.21812199074999694
},
"delivery": {
"n_samples": 58,
"mse": 0.5813553680199423,
"rmse": 0.7624666340371506,
"mae": 0.5806423342176552,
"r2": 0.26106224598641015
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6664637857577602,
"recall": 0.6949394155381325,
"f1": 0.672514619883041
},
"ee": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.6453567538126362,
"recall": 0.6947619047619048,
"f1": 0.6466727716727717
},
"toxic": {
"n_samples": 58,
"accuracy": 0.9655172413793104,
"precision": 0.75,
"recall": 0.9821428571428572,
"f1": 0.8242424242424242
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.30543536337189214,
"js_divergence": 0.055401807245651365
}
}

16
models/abl/both_biot5/20260624_015624/outer_fold_1/best_params.json Normal file
View File

@ -0,0 +1,16 @@
{
"dropout": 0.12715228141755225,
"lr": 0.00038210169217516986,
"weight_decay": 0.0032858044759097593,
"backbone_lr_ratio": 0.01707341286908778,
"moe_n_experts": 2,
"moe_top_k": 2,
"moe_expert_hidden_mult": 1,
"llm_lora_r": 16,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/both_biot5/20260624_015624/outer_fold_1/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 18}

420
models/abl/both_biot5/20260624_015624/outer_fold_1/history.json Normal file
View File

@ -0,0 +1,420 @@
{
"train": [
{
"loss": 5.880190706253051,
"loss_size": 1.2093870848417283,
"loss_pdi": 0.6751484453678132,
"loss_ee": 1.073475730419159,
"loss_delivery": 0.9780833125114441,
"loss_biodist": 1.262057900428772,
"loss_toxic": 0.6662091434001922,
"loss_moe_lb": 2.0
},
{
"loss": 5.294564938545227,
"loss_size": 1.1456821471452714,
"loss_pdi": 0.6367460191249847,
"loss_ee": 1.0255739212036132,
"loss_delivery": 0.8788796991109848,
"loss_biodist": 1.0146725714206695,
"loss_toxic": 0.5826438546180726,
"loss_moe_lb": 2.000000011920929
},
{
"loss": 4.824215388298034,
"loss_size": 1.0746640145778656,
"loss_pdi": 0.6190512955188752,
"loss_ee": 0.9799470067024231,
"loss_delivery": 0.8211173295974732,
"loss_biodist": 0.822171425819397,
"loss_toxic": 0.4998000741004944,
"loss_moe_lb": 2.000000023841858
},
{
"loss": 4.57146680355072,
"loss_size": 1.1148690223693847,
"loss_pdi": 0.5865873157978058,
"loss_ee": 0.933483499288559,
"loss_delivery": 0.7943474233150483,
"loss_biodist": 0.6780749917030334,
"loss_toxic": 0.4579812288284302,
"loss_moe_lb": 2.000000023841858
},
{
"loss": 4.209733653068542,
"loss_size": 1.004982291162014,
"loss_pdi": 0.5898078620433808,
"loss_ee": 0.8981360793113708,
"loss_delivery": 0.7758342236280441,
"loss_biodist": 0.5935669541358948,
"loss_toxic": 0.3401214301586151,
"loss_moe_lb": 1.999999988079071
},
{
"loss": 3.962087297439575,
"loss_size": 0.9623573675751687,
"loss_pdi": 0.5464575558900833,
"loss_ee": 0.883045095205307,
"loss_delivery": 0.7587750256061554,
"loss_biodist": 0.5211483776569367,
"loss_toxic": 0.2821056440472603,
"loss_moe_lb": 1.9999999761581422
},
{
"loss": 3.7815916776657104,
"loss_size": 0.9839702755212784,
"loss_pdi": 0.5576627612113952,
"loss_ee": 0.8658750712871551,
"loss_delivery": 0.6548837929964065,
"loss_biodist": 0.4737322092056274,
"loss_toxic": 0.23676925897598267,
"loss_moe_lb": 1.999999988079071
},
{
"loss": 3.6851258516311645,
"loss_size": 1.0055367261171342,
"loss_pdi": 0.5485720753669738,
"loss_ee": 0.8268381893634796,
"loss_delivery": 0.6720288872718811,
"loss_biodist": 0.39823555648326875,
"loss_toxic": 0.22523673437535763,
"loss_moe_lb": 2.0
},
{
"loss": 3.5697060585021974,
"loss_size": 0.9440896183252334,
"loss_pdi": 0.5229192078113556,
"loss_ee": 0.8265462815761566,
"loss_delivery": 0.6808992311358452,
"loss_biodist": 0.36781521439552306,
"loss_toxic": 0.21875333972275257,
"loss_moe_lb": 2.0
},
{
"loss": 3.3569907426834105,
"loss_size": 0.8516724839806556,
"loss_pdi": 0.5334105312824249,
"loss_ee": 0.8089183390140533,
"loss_delivery": 0.6359504997730255,
"loss_biodist": 0.3664983808994293,
"loss_toxic": 0.15149716436862945,
"loss_moe_lb": 1.999999964237213
},
{
"loss": 3.3709030628204344,
"loss_size": 0.8750916138291359,
"loss_pdi": 0.5084484338760376,
"loss_ee": 0.7743302106857299,
"loss_delivery": 0.7430924639105797,
"loss_biodist": 0.3230992153286934,
"loss_toxic": 0.13908824250102042,
"loss_moe_lb": 2.000000011920929
},
{
"loss": 3.3394760608673097,
"loss_size": 0.8576140075922012,
"loss_pdi": 0.5197807669639587,
"loss_ee": 0.7874584674835206,
"loss_delivery": 0.6742687806487083,
"loss_biodist": 0.3445627182722092,
"loss_toxic": 0.14837514758110046,
"loss_moe_lb": 1.999999988079071
},
{
"loss": 3.2558224201202393,
"loss_size": 0.8761242136359215,
"loss_pdi": 0.5131644994020462,
"loss_ee": 0.7845241308212281,
"loss_delivery": 0.6087752044200897,
"loss_biodist": 0.3249444216489792,
"loss_toxic": 0.14044311307370663,
"loss_moe_lb": 1.999999988079071
},
{
"loss": 3.3834808349609373,
"loss_size": 1.06574295014143,
"loss_pdi": 0.5020947188138962,
"loss_ee": 0.7692467987537384,
"loss_delivery": 0.6274533778429031,
"loss_biodist": 0.2866425186395645,
"loss_toxic": 0.12638874612748624,
"loss_moe_lb": 1.999999988079071
},
{
"loss": 3.1409812927246095,
"loss_size": 0.7952263683080674,
"loss_pdi": 0.5061811059713364,
"loss_ee": 0.7489465057849884,
"loss_delivery": 0.6558268785476684,
"loss_biodist": 0.297735820710659,
"loss_toxic": 0.12886845096945762,
"loss_moe_lb": 2.0
},
{
"loss": 3.073127257823944,
"loss_size": 0.810946398973465,
"loss_pdi": 0.482444629073143,
"loss_ee": 0.7382199764251709,
"loss_delivery": 0.6185100674629211,
"loss_biodist": 0.2800307020545006,
"loss_toxic": 0.13375547342002392,
"loss_moe_lb": 2.0
},
{
"loss": 3.3635661125183107,
"loss_size": 1.061921811103821,
"loss_pdi": 0.4904894888401031,
"loss_ee": 0.7482723414897918,
"loss_delivery": 0.6379371643066406,
"loss_biodist": 0.2663900047540665,
"loss_toxic": 0.15238383375108241,
"loss_moe_lb": 2.000000023841858
},
{
"loss": 3.229771542549133,
"loss_size": 0.9503776997327804,
"loss_pdi": 0.49906489849090574,
"loss_ee": 0.7416266202926636,
"loss_delivery": 0.584967689216137,
"loss_biodist": 0.2981968283653259,
"loss_toxic": 0.14783287178725005,
"loss_moe_lb": 1.9999999761581422
}
],
"val": [
{
"loss": 52.28438186645508,
"loss_size": 10.054548740386963,
"loss_pdi": 1.888471782207489,
"loss_ee": 2.3071884214878082,
"loss_delivery": 28.62615203857422,
"loss_biodist": 9.397465705871582,
"loss_toxic": 0.14978349208831787,
"loss_moe_lb": 2.0,
"acc_pdi": 0.36363636363636365,
"acc_ee": 0.30303030303030304,
"acc_toxic": 1.0
},
{
"loss": 19.082016944885254,
"loss_size": 1.859201729297638,
"loss_pdi": 0.8106305003166199,
"loss_ee": 1.5092724934220314,
"loss_delivery": 10.481206893920898,
"loss_biodist": 4.423018455505371,
"loss_toxic": 0.08147889375686646,
"loss_moe_lb": 2.0,
"acc_pdi": 0.36363636363636365,
"acc_ee": 0.5757575757575758,
"acc_toxic": 1.0
},
{
"loss": 8.121744871139526,
"loss_size": 0.36302029248327017,
"loss_pdi": 0.6240207403898239,
"loss_ee": 1.1844097077846527,
"loss_delivery": 3.6757107973098755,
"loss_biodist": 2.1130834817886353,
"loss_toxic": 0.19022316485643387,
"loss_moe_lb": 1.9999998807907104,
"acc_pdi": 0.6363636363636364,
"acc_ee": 0.5757575757575758,
"acc_toxic": 1.0
},
{
"loss": 4.076095461845398,
"loss_size": 0.503328412771225,
"loss_pdi": 0.6265619099140167,
"loss_ee": 0.8325532674789429,
"loss_delivery": 0.789628267288208,
"loss_biodist": 0.9311557412147522,
"loss_toxic": 0.38567738234996796,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.5757575757575758,
"acc_toxic": 1.0
},
{
"loss": 2.658109486103058,
"loss_size": 0.31043283944018185,
"loss_pdi": 0.609628364443779,
"loss_ee": 0.8723627924919128,
"loss_delivery": 0.13258208543993533,
"loss_biodist": 0.4401730000972748,
"loss_toxic": 0.2719712257385254,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.6666666666666666,
"acc_toxic": 1.0
},
{
"loss": 2.6200414896011353,
"loss_size": 0.3788522556424141,
"loss_pdi": 0.631707489490509,
"loss_ee": 0.9254117608070374,
"loss_delivery": 0.18661831319332123,
"loss_biodist": 0.326076865196228,
"loss_toxic": 0.15129034221172333,
"loss_moe_lb": 2.0,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.576258897781372,
"loss_size": 0.42240843176841736,
"loss_pdi": 0.6571086943149567,
"loss_ee": 0.8999056816101074,
"loss_delivery": 0.18188626319169998,
"loss_biodist": 0.3132934123277664,
"loss_toxic": 0.08257143944501877,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.4888967275619507,
"loss_size": 0.37972524762153625,
"loss_pdi": 0.6705151200294495,
"loss_ee": 0.8267186880111694,
"loss_delivery": 0.2078227400779724,
"loss_biodist": 0.338386669754982,
"loss_toxic": 0.04700608178973198,
"loss_moe_lb": 2.0,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.441374182701111,
"loss_size": 0.3871219903230667,
"loss_pdi": 0.639405757188797,
"loss_ee": 0.789948433637619,
"loss_delivery": 0.21282263845205307,
"loss_biodist": 0.3609292507171631,
"loss_toxic": 0.03286897484213114,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.310012698173523,
"loss_size": 0.36011902987957,
"loss_pdi": 0.6141173243522644,
"loss_ee": 0.7508799731731415,
"loss_delivery": 0.17297590523958206,
"loss_biodist": 0.3691263794898987,
"loss_toxic": 0.023637895472347736,
"loss_moe_lb": 2.0,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.246964752674103,
"loss_size": 0.3261938989162445,
"loss_pdi": 0.6195713728666306,
"loss_ee": 0.7365868091583252,
"loss_delivery": 0.1471389215439558,
"loss_biodist": 0.37987400591373444,
"loss_toxic": 0.018639557994902134,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.210842490196228,
"loss_size": 0.31696679443120956,
"loss_pdi": 0.6109910607337952,
"loss_ee": 0.7134979963302612,
"loss_delivery": 0.16695939376950264,
"loss_biodist": 0.36631646007299423,
"loss_toxic": 0.01727600209414959,
"loss_moe_lb": 2.0,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.1959953904151917,
"loss_size": 0.3190857395529747,
"loss_pdi": 0.6277008354663849,
"loss_ee": 0.7025340795516968,
"loss_delivery": 0.13767828233540058,
"loss_biodist": 0.37543031573295593,
"loss_toxic": 0.015050362795591354,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.2353000044822693,
"loss_size": 0.3203260600566864,
"loss_pdi": 0.6417901813983917,
"loss_ee": 0.7160685658454895,
"loss_delivery": 0.15194902382791042,
"loss_biodist": 0.37361618131399155,
"loss_toxic": 0.014141674619168043,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.2187429070472717,
"loss_size": 0.3165397047996521,
"loss_pdi": 0.6458726525306702,
"loss_ee": 0.6952449977397919,
"loss_delivery": 0.15103850327432156,
"loss_biodist": 0.38021840900182724,
"loss_toxic": 0.01233788300305605,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.2069762349128723,
"loss_size": 0.316790409386158,
"loss_pdi": 0.641569048166275,
"loss_ee": 0.689850926399231,
"loss_delivery": 0.15631976164877415,
"loss_biodist": 0.3722814545035362,
"loss_toxic": 0.01261688070371747,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.2091753482818604,
"loss_size": 0.32327602803707123,
"loss_pdi": 0.6411083936691284,
"loss_ee": 0.6842013895511627,
"loss_delivery": 0.1479108277708292,
"loss_biodist": 0.3834577724337578,
"loss_toxic": 0.011684997472912073,
"loss_moe_lb": 2.0,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
},
{
"loss": 2.20571631193161,
"loss_size": 0.31282903999090195,
"loss_pdi": 0.6427331268787384,
"loss_ee": 0.700422614812851,
"loss_delivery": 0.14261944591999054,
"loss_biodist": 0.37734580785036087,
"loss_toxic": 0.01235993904992938,
"loss_moe_lb": 1.9999999403953552,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.7272727272727273,
"acc_toxic": 1.0
}
]
}

1
models/abl/both_biot5/20260624_015624/outer_fold_1/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 3, 5, 6, 7, 10, 11, 12, 13, 15, 16, 18, 19, 20, 21, 24, 26, 27, 29, 30, 31, 32, 33, 35, 36, 38, 39, 40, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 65, 66, 67, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 97, 98, 99, 100, 101, 102, 103, 105, 106, 107, 108, 109, 110, 112, 114, 115, 118, 119, 120, 121, 123, 124, 125, 126, 128, 129, 130, 131, 132, 136, 137, 140, 141, 142, 143, 144, 145, 146, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 161, 162, 164, 165, 166, 167, 168, 169, 171, 172, 173, 174, 175, 176, 177, 178, 179, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 194, 195, 196, 197, 198, 199, 201, 202, 204, 205, 206, 207, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 225, 226, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 240, 242, 243, 244, 246, 247, 248, 250, 251, 252, 254, 255, 256, 258, 259, 260, 261, 262, 263, 264, 266, 268, 269, 270, 271, 273, 274, 275, 276, 277, 278, 279, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 295, 296, 297, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 313, 314, 315, 316, 317, 318, 319, 321, 322, 323, 324, 325, 327, 328, 329, 330, 331, 333, 334, 335, 337, 338, 339, 340, 341, 342, 343, 344, 345, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 361, 362, 364, 365, 366, 367, 368, 369, 373, 374, 375, 376, 378, 379, 380, 381, 382, 384, 387, 388, 389, 391, 392, 393, 394, 396, 397, 398, 399, 402, 403, 404, 405, 406, 408, 409, 410, 411, 412, 413, 414, 416, 417, 419], "outer_test_idx": [1, 2, 4, 8, 9, 14, 17, 22, 23, 25, 28, 34, 37, 41, 43, 45, 63, 64, 68, 85, 96, 104, 111, 113, 116, 117, 122, 127, 133, 134, 135, 138, 139, 147, 160, 163, 170, 180, 191, 192, 193, 200, 203, 208, 224, 227, 230, 239, 241, 245, 249, 253, 257, 265, 267, 272, 280, 294, 298, 299, 309, 312, 320, 326, 332, 336, 346, 347, 360, 363, 370, 371, 372, 377, 383, 385, 386, 390, 395, 400, 401, 407, 415, 418]}

42
models/abl/both_biot5/20260624_015624/outer_fold_1/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.3226448578157207,
"rmse": 0.5680183604565268,
"mae": 0.43127859149882125,
"r2": 0.014757311793952343
},
"delivery": {
"n_samples": 61,
"mse": 0.9731257240529179,
"rmse": 0.9864713498388678,
"mae": 0.6065680303717734,
"r2": 0.26341397848585624
},
"pdi": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.5783050847457627,
"recall": 0.5873015873015873,
"f1": 0.5808980755523877
},
"ee": {
"n_samples": 84,
"accuracy": 0.5238095238095238,
"precision": 0.4563973063973064,
"recall": 0.4560201703058846,
"f1": 0.4443340972752738
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9508196721311475,
"precision": 0.75,
"recall": 0.9741379310344828,
"f1": 0.8200589970501475
},
"biodist": {
"n_samples": 61,
"kl_divergence": 0.3699852731241905,
"js_divergence": 0.0921670637504171
}
}

16
models/abl/both_biot5/20260624_015624/outer_fold_2/best_params.json Normal file
View File

@ -0,0 +1,16 @@
{
"dropout": 0.33774333325981976,
"lr": 0.0008937362022799393,
"weight_decay": 0.0797179350056365,
"backbone_lr_ratio": 0.012019950560462185,
"moe_n_experts": 2,
"moe_top_k": 1,
"moe_expert_hidden_mult": 2,
"llm_lora_r": 8,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/both_biot5/20260624_015624/outer_fold_2/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 24}

558
models/abl/both_biot5/20260624_015624/outer_fold_2/history.json Normal file
View File

@ -0,0 +1,558 @@
{
"train": [
{
"loss": 5.5593267440795895,
"loss_size": 1.0105120539665222,
"loss_pdi": 0.7222382009029389,
"loss_ee": 1.0737396836280824,
"loss_delivery": 1.0050212740898132,
"loss_biodist": 1.1231000423431396,
"loss_toxic": 0.621865651011467,
"loss_moe_lb": 1.003492832183838
},
{
"loss": 4.787588620185852,
"loss_size": 0.8112421542406082,
"loss_pdi": 0.662821638584137,
"loss_ee": 1.0101438462734222,
"loss_delivery": 0.8974106043577195,
"loss_biodist": 0.8047115921974182,
"loss_toxic": 0.6087065398693084,
"loss_moe_lb": 1.0054036617279052
},
{
"loss": 4.188585686683655,
"loss_size": 0.783310429751873,
"loss_pdi": 0.645175290107727,
"loss_ee": 0.991130256652832,
"loss_delivery": 0.823707926273346,
"loss_biodist": 0.582040798664093,
"loss_toxic": 0.37240699082612994,
"loss_moe_lb": 1.0020507335662843
},
{
"loss": 3.9125982999801634,
"loss_size": 0.7660124808549881,
"loss_pdi": 0.6041824877262115,
"loss_ee": 0.9430283367633819,
"loss_delivery": 0.8412006556987762,
"loss_biodist": 0.5327713280916214,
"loss_toxic": 0.23650197237730025,
"loss_moe_lb": 1.0016139030456543
},
{
"loss": 3.7509905338287353,
"loss_size": 0.7014332443475724,
"loss_pdi": 0.5906592547893524,
"loss_ee": 0.9160506069660187,
"loss_delivery": 0.7715980872511864,
"loss_biodist": 0.48783472180366516,
"loss_toxic": 0.29581702277064326,
"loss_moe_lb": 1.001250147819519
},
{
"loss": 3.583024740219116,
"loss_size": 0.7506851077079773,
"loss_pdi": 0.5800322204828262,
"loss_ee": 0.8609747290611267,
"loss_delivery": 0.6995440624654293,
"loss_biodist": 0.4673841863870621,
"loss_toxic": 0.2362929627299309,
"loss_moe_lb": 1.0031494677066803
},
{
"loss": 3.4016853094100954,
"loss_size": 0.6849470436573029,
"loss_pdi": 0.5577913045883178,
"loss_ee": 0.8802375853061676,
"loss_delivery": 0.711587107181549,
"loss_biodist": 0.3890023499727249,
"loss_toxic": 0.1908457398414612,
"loss_moe_lb": 1.0012832641601563
},
{
"loss": 3.3591570615768434,
"loss_size": 0.7104779332876205,
"loss_pdi": 0.537405812740326,
"loss_ee": 0.8152894020080567,
"loss_delivery": 0.7708096042275429,
"loss_biodist": 0.39673206508159636,
"loss_toxic": 0.14370608888566494,
"loss_moe_lb": 1.002393662929535
},
{
"loss": 3.2598351001739503,
"loss_size": 0.6960233926773072,
"loss_pdi": 0.5244608700275422,
"loss_ee": 0.8347655832767487,
"loss_delivery": 0.7051882833242417,
"loss_biodist": 0.3773863077163696,
"loss_toxic": 0.13903691042214633,
"loss_moe_lb": 1.0038196325302124
},
{
"loss": 3.131180429458618,
"loss_size": 0.6922156900167465,
"loss_pdi": 0.5219623118638992,
"loss_ee": 0.8243920862674713,
"loss_delivery": 0.6285123139619827,
"loss_biodist": 0.32298070192337036,
"loss_toxic": 0.15709903929382563,
"loss_moe_lb": 1.001521611213684
},
{
"loss": 3.0642634630203247,
"loss_size": 0.6980895549058914,
"loss_pdi": 0.5265205204486847,
"loss_ee": 0.7623060345649719,
"loss_delivery": 0.6771705731749534,
"loss_biodist": 0.29747152924537656,
"loss_toxic": 0.12104640826582909,
"loss_moe_lb": 1.0047925055027007
},
{
"loss": 2.9357067823410032,
"loss_size": 0.6488922163844109,
"loss_pdi": 0.5127476632595063,
"loss_ee": 0.7586858332157135,
"loss_delivery": 0.603175763785839,
"loss_biodist": 0.31603043228387834,
"loss_toxic": 0.11426987387239933,
"loss_moe_lb": 1.0011218011379241
},
{
"loss": 3.0185377836227416,
"loss_size": 0.6770095571875572,
"loss_pdi": 0.50152867436409,
"loss_ee": 0.7681885123252868,
"loss_delivery": 0.6976583883166313,
"loss_biodist": 0.28682698458433153,
"loss_toxic": 0.11139093395322561,
"loss_moe_lb": 1.002818661928177
},
{
"loss": 3.0447335243225098,
"loss_size": 0.6448296248912812,
"loss_pdi": 0.506335175037384,
"loss_ee": 0.791974276304245,
"loss_delivery": 0.7140580207109452,
"loss_biodist": 0.2721629112958908,
"loss_toxic": 0.1418131623417139,
"loss_moe_lb": 1.0018205165863037
},
{
"loss": 2.7945156335830688,
"loss_size": 0.592767271399498,
"loss_pdi": 0.5045191138982773,
"loss_ee": 0.7424894750118256,
"loss_delivery": 0.6036672025918961,
"loss_biodist": 0.26116000711917875,
"loss_toxic": 0.11251284927129745,
"loss_moe_lb": 1.0046443819999695
},
{
"loss": 2.9158809781074524,
"loss_size": 0.6414759591221809,
"loss_pdi": 0.49571222364902495,
"loss_ee": 0.7354807734489441,
"loss_delivery": 0.6598402425646782,
"loss_biodist": 0.2649649605154991,
"loss_toxic": 0.1430682374164462,
"loss_moe_lb": 1.0039978861808776
},
{
"loss": 2.7685218334197996,
"loss_size": 0.5974807739257812,
"loss_pdi": 0.48887917399406433,
"loss_ee": 0.7438495635986329,
"loss_delivery": 0.5861493825912476,
"loss_biodist": 0.2618252247571945,
"loss_toxic": 0.11498904321342707,
"loss_moe_lb": 1.0042200565338135
},
{
"loss": 2.677497720718384,
"loss_size": 0.5983023896813393,
"loss_pdi": 0.46180991232395174,
"loss_ee": 0.7298104524612427,
"loss_delivery": 0.5947681233286858,
"loss_biodist": 0.23806397169828414,
"loss_toxic": 0.0808176543097943,
"loss_moe_lb": 1.0028916001319885
},
{
"loss": 2.7354037761688232,
"loss_size": 0.5406947076320648,
"loss_pdi": 0.48281011581420896,
"loss_ee": 0.7385894417762756,
"loss_delivery": 0.6557572320103645,
"loss_biodist": 0.24288078397512436,
"loss_toxic": 0.10278778467327357,
"loss_moe_lb": 1.0018792033195496
},
{
"loss": 2.8157411932945253,
"loss_size": 0.5629876017570495,
"loss_pdi": 0.5092323899269104,
"loss_ee": 0.7621454715728759,
"loss_delivery": 0.6139041751623153,
"loss_biodist": 0.2855973556637764,
"loss_toxic": 0.11170732360333205,
"loss_moe_lb": 1.000916564464569
},
{
"loss": 2.765420413017273,
"loss_size": 0.5669456720352173,
"loss_pdi": 0.4541279703378677,
"loss_ee": 0.7568132042884826,
"loss_delivery": 0.633655896782875,
"loss_biodist": 0.2770042195916176,
"loss_toxic": 0.10590465851128102,
"loss_moe_lb": 0.9998559474945068
},
{
"loss": 2.6403432369232176,
"loss_size": 0.5567607164382935,
"loss_pdi": 0.46041806042194366,
"loss_ee": 0.7157784104347229,
"loss_delivery": 0.5941496953368187,
"loss_biodist": 0.2570425972342491,
"loss_toxic": 0.08308861032128334,
"loss_moe_lb": 1.000298011302948
},
{
"loss": 2.639028179645538,
"loss_size": 0.4794127196073532,
"loss_pdi": 0.47615211009979247,
"loss_ee": 0.7287471771240235,
"loss_delivery": 0.6262050732970238,
"loss_biodist": 0.24853375554084778,
"loss_toxic": 0.10661754999309778,
"loss_moe_lb": 1.0017034769058228
},
{
"loss": 2.763648736476898,
"loss_size": 0.5313900172710418,
"loss_pdi": 0.4860714554786682,
"loss_ee": 0.717977887392044,
"loss_delivery": 0.6682430401444435,
"loss_biodist": 0.24781301766633987,
"loss_toxic": 0.13892789240926504,
"loss_moe_lb": 1.003730058670044
}
],
"val": [
{
"loss": 335.33447265625,
"loss_size": 281.84617614746094,
"loss_pdi": 5.466546177864075,
"loss_ee": 6.976545572280884,
"loss_delivery": 38.42522048950195,
"loss_biodist": 3.7248597145080566,
"loss_toxic": 1.3795287486573216e-05,
"loss_moe_lb": 1.9999961853027344,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.24242424242424243,
"acc_toxic": 1.0
},
{
"loss": 128.69612503051758,
"loss_size": 113.34328842163086,
"loss_pdi": 2.330305337905884,
"loss_ee": 4.655527472496033,
"loss_delivery": 6.972789287567139,
"loss_biodist": 2.3381829261779785,
"loss_toxic": 0.0002481710980646312,
"loss_moe_lb": 1.9992881417274475,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5151515151515151,
"acc_toxic": 1.0
},
{
"loss": 37.53413009643555,
"loss_size": 30.36787700653076,
"loss_pdi": 1.507568359375,
"loss_ee": 2.8481683135032654,
"loss_delivery": 2.3734264373779297,
"loss_biodist": 0.7999420762062073,
"loss_toxic": 0.017548002302646637,
"loss_moe_lb": 1.9912574887275696,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5757575757575758,
"acc_toxic": 1.0
},
{
"loss": 12.071053981781006,
"loss_size": 7.413588523864746,
"loss_pdi": 1.51066455245018,
"loss_ee": 1.9464640021324158,
"loss_delivery": 1.0180751085281372,
"loss_biodist": 0.215897798538208,
"loss_toxic": 0.10842924565076828,
"loss_moe_lb": 1.9459542036056519,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 7.088524580001831,
"loss_size": 3.7983028888702393,
"loss_pdi": 1.0122159123420715,
"loss_ee": 1.1055640280246735,
"loss_delivery": 0.9112273454666138,
"loss_biodist": 0.19447730481624603,
"loss_toxic": 0.13830989599227905,
"loss_moe_lb": 1.7341642379760742,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.95
},
{
"loss": 6.754403829574585,
"loss_size": 3.893798828125,
"loss_pdi": 0.7364282011985779,
"loss_ee": 0.9880420565605164,
"loss_delivery": 0.9043927192687988,
"loss_biodist": 0.19323037564754486,
"loss_toxic": 0.10924768447875977,
"loss_moe_lb": 1.2978886365890503,
"acc_pdi": 0.5151515151515151,
"acc_ee": 0.5454545454545454,
"acc_toxic": 0.95
},
{
"loss": 7.01631498336792,
"loss_size": 4.065897583961487,
"loss_pdi": 0.7407042682170868,
"loss_ee": 1.0901811718940735,
"loss_delivery": 0.9028797149658203,
"loss_biodist": 0.1832374930381775,
"loss_toxic": 0.11952021718025208,
"loss_moe_lb": 1.0571035742759705,
"acc_pdi": 0.5151515151515151,
"acc_ee": 0.5454545454545454,
"acc_toxic": 0.95
},
{
"loss": 6.703785419464111,
"loss_size": 3.783825993537903,
"loss_pdi": 0.741980105638504,
"loss_ee": 1.1159003674983978,
"loss_delivery": 0.8602625131607056,
"loss_biodist": 0.18989908695220947,
"loss_toxic": 0.10079815238714218,
"loss_moe_lb": 1.0017997920513153,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.95
},
{
"loss": 6.863262176513672,
"loss_size": 3.8816100358963013,
"loss_pdi": 0.7432284653186798,
"loss_ee": 1.1886624097824097,
"loss_delivery": 0.882551372051239,
"loss_biodist": 0.1726650446653366,
"loss_toxic": 0.09364569187164307,
"loss_moe_lb": 0.9985060095787048,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.95
},
{
"loss": 6.566705703735352,
"loss_size": 3.700358748435974,
"loss_pdi": 0.7168927788734436,
"loss_ee": 1.088432788848877,
"loss_delivery": 0.8829938769340515,
"loss_biodist": 0.1824142187833786,
"loss_toxic": 0.09188876301050186,
"loss_moe_lb": 1.003427267074585,
"acc_pdi": 0.6363636363636364,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.493593454360962,
"loss_size": 3.410131812095642,
"loss_pdi": 0.7412647008895874,
"loss_ee": 1.3382400870323181,
"loss_delivery": 0.8545717597007751,
"loss_biodist": 0.16492798924446106,
"loss_toxic": 0.09331875294446945,
"loss_moe_lb": 0.9975298047065735,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.5454545454545454,
"acc_toxic": 0.95
},
{
"loss": 6.1560218334198,
"loss_size": 3.42904794216156,
"loss_pdi": 0.6950583755970001,
"loss_ee": 1.0537507236003876,
"loss_delivery": 0.8217563629150391,
"loss_biodist": 0.1508345603942871,
"loss_toxic": 0.10079897195100784,
"loss_moe_lb": 1.0026240944862366,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.1669135093688965,
"loss_size": 3.4153796434402466,
"loss_pdi": 0.7046009600162506,
"loss_ee": 1.1063185334205627,
"loss_delivery": 0.8078489303588867,
"loss_biodist": 0.14892606437206268,
"loss_toxic": 0.08529356867074966,
"loss_moe_lb": 0.9994112849235535,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.088647842407227,
"loss_size": 3.3628441095352173,
"loss_pdi": 0.6925024688243866,
"loss_ee": 1.095509558916092,
"loss_delivery": 0.8118428587913513,
"loss_biodist": 0.13240604102611542,
"loss_toxic": 0.09537149220705032,
"loss_moe_lb": 1.0020636916160583,
"acc_pdi": 0.5757575757575758,
"acc_ee": 0.5454545454545454,
"acc_toxic": 0.95
},
{
"loss": 6.245176792144775,
"loss_size": 3.4187978506088257,
"loss_pdi": 0.7559232711791992,
"loss_ee": 1.1447378993034363,
"loss_delivery": 0.8184512257575989,
"loss_biodist": 0.13180184364318848,
"loss_toxic": 0.08590542525053024,
"loss_moe_lb": 0.9998006224632263,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.5454545454545454,
"acc_toxic": 0.95
},
{
"loss": 6.650262832641602,
"loss_size": 3.4030256271362305,
"loss_pdi": 0.78317990899086,
"loss_ee": 1.57785964012146,
"loss_delivery": 0.8096794486045837,
"loss_biodist": 0.1302957981824875,
"loss_toxic": 0.08302690088748932,
"loss_moe_lb": 1.0095041990280151,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.5454545454545454,
"acc_toxic": 0.95
},
{
"loss": 6.310943841934204,
"loss_size": 3.4086145162582397,
"loss_pdi": 0.780304342508316,
"loss_ee": 1.2179460525512695,
"loss_delivery": 0.8235325217247009,
"loss_biodist": 0.12248428910970688,
"loss_toxic": 0.07666709274053574,
"loss_moe_lb": 1.003424346446991,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.5151515151515151,
"acc_toxic": 0.95
},
{
"loss": 6.448458433151245,
"loss_size": 3.31357479095459,
"loss_pdi": 0.7256665527820587,
"loss_ee": 1.5409347414970398,
"loss_delivery": 0.8057543039321899,
"loss_biodist": 0.12350723892450333,
"loss_toxic": 0.07301509380340576,
"loss_moe_lb": 1.0029084086418152,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.1759679317474365,
"loss_size": 3.315897583961487,
"loss_pdi": 0.7544052302837372,
"loss_ee": 1.2371982038021088,
"loss_delivery": 0.795327365398407,
"loss_biodist": 0.11834156513214111,
"loss_toxic": 0.07248509675264359,
"loss_moe_lb": 1.0009818375110626,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.230125427246094,
"loss_size": 3.3428906202316284,
"loss_pdi": 0.7695878148078918,
"loss_ee": 1.2665679454803467,
"loss_delivery": 0.788640558719635,
"loss_biodist": 0.11356551945209503,
"loss_toxic": 0.06959143280982971,
"loss_moe_lb": 1.001686692237854,
"acc_pdi": 0.6363636363636364,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.286802530288696,
"loss_size": 3.384185552597046,
"loss_pdi": 0.7634362280368805,
"loss_ee": 1.28605717420578,
"loss_delivery": 0.7914952635765076,
"loss_biodist": 0.11232580244541168,
"loss_toxic": 0.07194364815950394,
"loss_moe_lb": 0.9993330836296082,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.5454545454545454,
"acc_toxic": 0.95
},
{
"loss": 6.318703889846802,
"loss_size": 3.407394051551819,
"loss_pdi": 0.7701711654663086,
"loss_ee": 1.2932927906513214,
"loss_delivery": 0.7890445590019226,
"loss_biodist": 0.11213445663452148,
"loss_toxic": 0.07041720300912857,
"loss_moe_lb": 1.0023659467697144,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.5757575757575758,
"acc_toxic": 0.95
},
{
"loss": 6.275263547897339,
"loss_size": 3.352391481399536,
"loss_pdi": 0.7932782769203186,
"loss_ee": 1.2884126007556915,
"loss_delivery": 0.7835452556610107,
"loss_biodist": 0.10959138721227646,
"loss_toxic": 0.07115987688302994,
"loss_moe_lb": 1.00303316116333,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5151515151515151,
"acc_toxic": 0.95
},
{
"loss": 6.3098509311676025,
"loss_size": 3.3641092777252197,
"loss_pdi": 0.7913138270378113,
"loss_ee": 1.2944900691509247,
"loss_delivery": 0.7982775568962097,
"loss_biodist": 0.11419817060232162,
"loss_toxic": 0.07155168801546097,
"loss_moe_lb": 1.0040580034255981,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5151515151515151,
"acc_toxic": 0.95
}
]
}

1
models/abl/both_biot5/20260624_015624/outer_fold_2/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 13, 14, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 32, 33, 34, 36, 37, 38, 39, 40, 41, 43, 44, 45, 47, 48, 49, 50, 51, 55, 56, 57, 58, 59, 60, 61, 63, 64, 65, 68, 69, 70, 72, 74, 76, 78, 79, 80, 82, 83, 84, 85, 86, 87, 88, 91, 93, 94, 95, 96, 97, 100, 101, 102, 103, 104, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 142, 143, 145, 147, 148, 149, 150, 151, 152, 153, 156, 157, 158, 159, 160, 161, 162, 163, 165, 166, 167, 168, 170, 171, 172, 173, 174, 175, 177, 179, 180, 181, 183, 184, 185, 186, 187, 188, 191, 192, 193, 194, 196, 197, 199, 200, 202, 203, 204, 205, 207, 208, 209, 210, 211, 212, 213, 214, 215, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 232, 233, 235, 236, 237, 238, 239, 240, 241, 242, 243, 245, 246, 249, 250, 251, 252, 253, 254, 255, 256, 257, 259, 260, 261, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 280, 281, 282, 283, 284, 285, 287, 289, 290, 293, 294, 295, 296, 297, 298, 299, 300, 302, 303, 305, 307, 308, 309, 310, 311, 312, 313, 314, 315, 317, 318, 320, 321, 322, 323, 324, 326, 327, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 358, 360, 361, 363, 364, 365, 366, 368, 369, 370, 371, 372, 373, 376, 377, 378, 379, 380, 381, 382, 383, 385, 386, 387, 388, 389, 390, 392, 394, 395, 398, 399, 400, 401, 402, 404, 405, 407, 408, 409, 410, 412, 413, 414, 415, 416, 417, 418], "outer_test_idx": [10, 12, 15, 31, 35, 42, 46, 52, 53, 54, 62, 66, 67, 71, 73, 75, 77, 81, 89, 90, 92, 98, 99, 105, 123, 124, 141, 144, 146, 154, 155, 164, 169, 176, 178, 182, 189, 190, 195, 198, 201, 206, 216, 217, 218, 219, 231, 234, 244, 247, 248, 258, 262, 270, 277, 278, 279, 286, 288, 291, 292, 301, 304, 306, 316, 319, 325, 328, 343, 357, 359, 362, 367, 374, 375, 384, 391, 393, 396, 397, 403, 406, 411, 419]}

42
models/abl/both_biot5/20260624_015624/outer_fold_2/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 84,
"mse": 0.43468907454487105,
"rmse": 0.6593095437993227,
"mae": 0.5041968602287982,
"r2": 0.14149684434925025
},
"delivery": {
"n_samples": 60,
"mse": 0.6609372574877708,
"rmse": 0.8129804779253748,
"mae": 0.5105325942936664,
"r2": 0.12017795405995413
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6592592592592592,
"recall": 0.6891495601173021,
"f1": 0.6657824933687002
},
"ee": {
"n_samples": 84,
"accuracy": 0.6785714285714286,
"precision": 0.6277327935222673,
"recall": 0.6944444444444443,
"f1": 0.6407632082237629
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9344262295081968,
"precision": 0.7142857142857143,
"recall": 0.9655172413793103,
"f1": 0.7821428571428571
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.2188401979219755,
"js_divergence": 0.052433191262446874
}
}

16
models/abl/both_biot5/20260624_015624/outer_fold_3/best_params.json Normal file
View File

@ -0,0 +1,16 @@
{
"dropout": 0.10301838844131589,
"lr": 0.0008648998582653286,
"weight_decay": 0.0011644667702360517,
"backbone_lr_ratio": 0.9869864342312137,
"moe_n_experts": 4,
"moe_top_k": 2,
"moe_expert_hidden_mult": 2,
"llm_lora_r": 32,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/both_biot5/20260624_015624/outer_fold_3/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 15}

350
models/abl/both_biot5/20260624_015624/outer_fold_3/history.json Normal file
View File

@ -0,0 +1,350 @@
{
"train": [
{
"loss": 4.98353123664856,
"loss_size": 0.5103443682193756,
"loss_pdi": 0.6624919235706329,
"loss_ee": 1.0547867655754088,
"loss_delivery": 1.056617796421051,
"loss_biodist": 1.086611419916153,
"loss_toxic": 0.596496444940567,
"loss_moe_lb": 2.2669289350509643
},
{
"loss": 3.8003307819366454,
"loss_size": 0.4525773376226425,
"loss_pdi": 0.5971371352672576,
"loss_ee": 0.9487643420696259,
"loss_delivery": 0.7854501068592071,
"loss_biodist": 0.6455104947090149,
"loss_toxic": 0.3577138617634773,
"loss_moe_lb": 2.2061395168304445
},
{
"loss": 3.303273916244507,
"loss_size": 0.433304725587368,
"loss_pdi": 0.5875962555408478,
"loss_ee": 0.8717754185199738,
"loss_delivery": 0.7809680938720703,
"loss_biodist": 0.4639909386634827,
"loss_toxic": 0.15247682817280292,
"loss_moe_lb": 2.2805288076400756
},
{
"loss": 3.526792788505554,
"loss_size": 0.4987005777657032,
"loss_pdi": 0.570679122209549,
"loss_ee": 0.936754584312439,
"loss_delivery": 0.9002981066703797,
"loss_biodist": 0.3439763605594635,
"loss_toxic": 0.27062796857208016,
"loss_moe_lb": 2.2777772903442384
},
{
"loss": 3.268141341209412,
"loss_size": 0.5438169747591018,
"loss_pdi": 0.5937646687030792,
"loss_ee": 0.9139786481857299,
"loss_delivery": 0.8525178879499435,
"loss_biodist": 0.23290605545043946,
"loss_toxic": 0.12935640322975814,
"loss_moe_lb": 2.22677104473114
},
{
"loss": 2.7174599885940554,
"loss_size": 0.3811827048659325,
"loss_pdi": 0.543531545996666,
"loss_ee": 0.7985063374042511,
"loss_delivery": 0.6488073140382766,
"loss_biodist": 0.1903185188770294,
"loss_toxic": 0.14885550662875174,
"loss_moe_lb": 2.2154770135879516
},
{
"loss": 2.7810311913490295,
"loss_size": 0.4000975012779236,
"loss_pdi": 0.5123270153999329,
"loss_ee": 0.7565424919128418,
"loss_delivery": 0.8099995672702789,
"loss_biodist": 0.213769643753767,
"loss_toxic": 0.08841842513065785,
"loss_moe_lb": 2.2401757717132567
},
{
"loss": 2.7185701966285705,
"loss_size": 0.44551015496253965,
"loss_pdi": 0.49670147001743314,
"loss_ee": 0.73055379986763,
"loss_delivery": 0.5621314287185669,
"loss_biodist": 0.17535570934414862,
"loss_toxic": 0.2991587908938527,
"loss_moe_lb": 2.2781644821166993
},
{
"loss": 2.2888001203536987,
"loss_size": 0.3394672155380249,
"loss_pdi": 0.48539329767227174,
"loss_ee": 0.7185316145420074,
"loss_delivery": 0.5121426343917846,
"loss_biodist": 0.13291144520044326,
"loss_toxic": 0.09854903249070049,
"loss_moe_lb": 2.284846830368042
},
{
"loss": 2.207217276096344,
"loss_size": 0.3658413976430893,
"loss_pdi": 0.46658535301685333,
"loss_ee": 0.6700418382883072,
"loss_delivery": 0.5210796371102333,
"loss_biodist": 0.12452970072627068,
"loss_toxic": 0.0588332430459559,
"loss_moe_lb": 2.3118534803390505
},
{
"loss": 2.286478888988495,
"loss_size": 0.5112962543964386,
"loss_pdi": 0.4537936896085739,
"loss_ee": 0.6289072722196579,
"loss_delivery": 0.507325604557991,
"loss_biodist": 0.1407230131328106,
"loss_toxic": 0.043172567570582035,
"loss_moe_lb": 2.3088854789733886
},
{
"loss": 1.9082700490951539,
"loss_size": 0.3032146096229553,
"loss_pdi": 0.4398723304271698,
"loss_ee": 0.5828634619712829,
"loss_delivery": 0.42030360326170924,
"loss_biodist": 0.12580163292586805,
"loss_toxic": 0.03535112321842462,
"loss_moe_lb": 2.323744511604309
},
{
"loss": 1.8958350777626038,
"loss_size": 0.3045516416430473,
"loss_pdi": 0.42868764102458956,
"loss_ee": 0.6063023418188095,
"loss_delivery": 0.3859559893608093,
"loss_biodist": 0.12728828862309455,
"loss_toxic": 0.043184947408735755,
"loss_moe_lb": 2.3214282274246214
},
{
"loss": 1.8829688549041748,
"loss_size": 0.2849637731909752,
"loss_pdi": 0.43566767275333407,
"loss_ee": 0.5696669220924377,
"loss_delivery": 0.43352929055690764,
"loss_biodist": 0.12172005996108055,
"loss_toxic": 0.03850965294986963,
"loss_moe_lb": 2.3120303392410277
},
{
"loss": 1.8242438197135926,
"loss_size": 0.2617476776242256,
"loss_pdi": 0.4145724028348923,
"loss_ee": 0.5960349798202514,
"loss_delivery": 0.38009128868579867,
"loss_biodist": 0.11558513827621937,
"loss_toxic": 0.0566928680986166,
"loss_moe_lb": 2.314754915237427
}
],
"val": [
{
"loss": 24.005695343017578,
"loss_size": 15.54491138458252,
"loss_pdi": 1.2003737390041351,
"loss_ee": 2.2405465245246887,
"loss_delivery": 1.5608795881271362,
"loss_biodist": 3.4996793270111084,
"loss_moe_lb": 4.0,
"acc_pdi": 0.24242424242424243,
"acc_ee": 0.6666666666666666,
"acc_toxic": 1.0
},
{
"loss": 8.060338735580444,
"loss_size": 3.58082914352417,
"loss_pdi": 1.2808009386062622,
"loss_ee": 1.1585214138031006,
"loss_delivery": 1.1900737285614014,
"loss_biodist": 0.8504003882408142,
"loss_toxic": 1.5549039389384234e-08,
"loss_moe_lb": 3.9999998807907104,
"acc_pdi": 0.24242424242424243,
"acc_ee": 0.6666666666666666,
"acc_toxic": 1.0
},
{
"loss": 4.360226154327393,
"loss_size": 1.4595853378996253,
"loss_pdi": 0.6863298714160919,
"loss_ee": 0.9662824869155884,
"loss_delivery": 0.7637103796005249,
"loss_biodist": 0.4697404205799103,
"loss_toxic": 8.277022061520256e-06,
"loss_moe_lb": 3.999418258666992,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6363636363636364,
"acc_toxic": 1.0
},
{
"loss": 4.396560609340668,
"loss_size": 1.8572371751070023,
"loss_pdi": 0.5949408262968063,
"loss_ee": 0.7665933966636658,
"loss_delivery": 0.8252537250518799,
"loss_biodist": 0.33583864569664,
"loss_toxic": 0.0002622886386234313,
"loss_moe_lb": 3.7502208948135376,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.5151515151515151,
"acc_toxic": 1.0
},
{
"loss": 3.7490410208702087,
"loss_size": 1.533302428200841,
"loss_pdi": 0.5374033600091934,
"loss_ee": 0.796456903219223,
"loss_delivery": 0.6736778616905212,
"loss_biodist": 0.1171751543879509,
"loss_toxic": 0.07388537377119064,
"loss_moe_lb": 3.304153323173523,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.7017577290534973,
"loss_size": 1.452047474682331,
"loss_pdi": 0.5511361062526703,
"loss_ee": 0.7534093856811523,
"loss_delivery": 0.5486688017845154,
"loss_biodist": 0.10727711766958237,
"loss_toxic": 0.27184852957725525,
"loss_moe_lb": 2.7834482192993164,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.3687055110931396,
"loss_size": 1.48643309623003,
"loss_pdi": 0.3446184918284416,
"loss_ee": 0.6335599273443222,
"loss_delivery": 0.6432744264602661,
"loss_biodist": 0.08594726771116257,
"loss_toxic": 0.1547670066356659,
"loss_moe_lb": 2.3777592182159424,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.9130434782608695
},
{
"loss": 2.87318617105484,
"loss_size": 1.189874367788434,
"loss_pdi": 0.3619840368628502,
"loss_ee": 0.6468346267938614,
"loss_delivery": 0.40307384729385376,
"loss_biodist": 0.055318526923656464,
"loss_toxic": 0.19129060208797455,
"loss_moe_lb": 2.4826799631118774,
"acc_pdi": 0.8787878787878788,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.201488822698593,
"loss_size": 1.4587422758340836,
"loss_pdi": 0.38734911382198334,
"loss_ee": 0.594419464468956,
"loss_delivery": 0.4994201362133026,
"loss_biodist": 0.06301930546760559,
"loss_toxic": 0.17198005318641663,
"loss_moe_lb": 2.542781352996826,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.9130434782608695
},
{
"loss": 2.99306982755661,
"loss_size": 1.215458706021309,
"loss_pdi": 0.3945038765668869,
"loss_ee": 0.6303490996360779,
"loss_delivery": 0.4833860993385315,
"loss_biodist": 0.05768001824617386,
"loss_toxic": 0.1879633069038391,
"loss_moe_lb": 2.5431123971939087,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.253591477870941,
"loss_size": 1.5342472940683365,
"loss_pdi": 0.3926478773355484,
"loss_ee": 0.62346450984478,
"loss_delivery": 0.48901209235191345,
"loss_biodist": 0.06029331684112549,
"loss_toxic": 0.12806108593940735,
"loss_moe_lb": 2.5561182498931885,
"acc_pdi": 0.8181818181818182,
"acc_ee": 0.696969696969697,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.1617013812065125,
"loss_size": 1.4483742266893387,
"loss_pdi": 0.37433257699012756,
"loss_ee": 0.6179856806993484,
"loss_delivery": 0.48835670948028564,
"loss_biodist": 0.05660872906446457,
"loss_toxic": 0.14916129410266876,
"loss_moe_lb": 2.591035842895508,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.0851848125457764,
"loss_size": 1.424731507897377,
"loss_pdi": 0.363551527261734,
"loss_ee": 0.6133799254894257,
"loss_delivery": 0.44111400842666626,
"loss_biodist": 0.06433821469545364,
"loss_toxic": 0.14992277324199677,
"loss_moe_lb": 2.562776565551758,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.051860749721527,
"loss_size": 1.3775647431612015,
"loss_pdi": 0.37555189430713654,
"loss_ee": 0.6089565008878708,
"loss_delivery": 0.4281175136566162,
"loss_biodist": 0.06453074514865875,
"loss_toxic": 0.1686825305223465,
"loss_moe_lb": 2.5477943420410156,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.8695652173913043
},
{
"loss": 3.0738858580589294,
"loss_size": 1.3745263665914536,
"loss_pdi": 0.37682630121707916,
"loss_ee": 0.6088242530822754,
"loss_delivery": 0.4551859498023987,
"loss_biodist": 0.06464586406946182,
"loss_toxic": 0.16618046164512634,
"loss_moe_lb": 2.543630838394165,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6666666666666666,
"acc_toxic": 0.8695652173913043
}
]
}

1
models/abl/both_biot5/20260624_015624/outer_fold_3/splits.json Normal file
View File

@ -0,0 +1 @@
{"outer_train_idx": [0, 1, 2, 4, 5, 7, 8, 9, 10, 11, 12, 14, 15, 16, 17, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 32, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 45, 46, 47, 51, 52, 53, 54, 55, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 73, 74, 75, 76, 77, 78, 79, 81, 82, 83, 84, 85, 86, 87, 89, 90, 91, 92, 93, 95, 96, 97, 98, 99, 100, 102, 104, 105, 106, 109, 110, 111, 112, 113, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 127, 129, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 146, 147, 148, 150, 151, 152, 153, 154, 155, 156, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 172, 173, 174, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 189, 190, 191, 192, 193, 194, 195, 198, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 212, 213, 214, 215, 216, 217, 218, 219, 221, 223, 224, 226, 227, 228, 230, 231, 233, 234, 235, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 251, 252, 253, 254, 255, 256, 257, 258, 259, 262, 263, 264, 265, 266, 267, 268, 269, 270, 272, 273, 274, 275, 276, 277, 278, 279, 280, 282, 284, 286, 288, 289, 290, 291, 292, 293, 294, 295, 297, 298, 299, 301, 302, 304, 305, 306, 307, 308, 309, 310, 312, 314, 315, 316, 317, 319, 320, 321, 325, 326, 327, 328, 331, 332, 334, 336, 340, 341, 343, 344, 346, 347, 348, 350, 352, 353, 357, 359, 360, 361, 362, 363, 364, 366, 367, 368, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 382, 383, 384, 385, 386, 387, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 400, 401, 402, 403, 404, 406, 407, 410, 411, 412, 414, 415, 416, 418, 419], "outer_test_idx": [3, 6, 13, 18, 19, 26, 33, 44, 48, 49, 50, 56, 57, 72, 80, 88, 94, 101, 103, 107, 108, 114, 126, 128, 130, 145, 149, 157, 167, 171, 175, 188, 196, 197, 199, 211, 220, 222, 225, 229, 232, 236, 250, 260, 261, 271, 281, 283, 285, 287, 296, 300, 303, 311, 313, 318, 322, 323, 324, 329, 330, 333, 335, 337, 338, 339, 342, 345, 349, 351, 354, 355, 356, 358, 365, 369, 381, 388, 399, 405, 408, 409, 413, 417]}

42
models/abl/both_biot5/20260624_015624/outer_fold_3/test_metrics.json Normal file
View File

@ -0,0 +1,42 @@
{
"size": {
"n_samples": 83,
"mse": 1.550640342858784,
"rmse": 1.245247101124425,
"mae": 0.5501430786989001,
"r2": 0.12294600324587124
},
"delivery": {
"n_samples": 59,
"mse": 0.748405480052253,
"rmse": 0.8651043174393785,
"mae": 0.5687398803657142,
"r2": 0.2724609546561282
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6445623342175066,
"recall": 0.6598240469208211,
"f1": 0.65
},
"ee": {
"n_samples": 84,
"accuracy": 0.6428571428571429,
"precision": 0.5892509050403788,
"recall": 0.6273504273504273,
"f1": 0.5933913253594297
},
"toxic": {
"n_samples": 60,
"accuracy": 0.9666666666666667,
"precision": 0.8,
"recall": 0.9824561403508771,
"f1": 0.8660714285714286
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.16217964699830698,
"js_divergence": 0.0385441230143625
}
}

16
models/abl/both_biot5/20260624_015624/outer_fold_4/best_params.json Normal file
View File

@ -0,0 +1,16 @@
{
"dropout": 0.303142745975858,
"lr": 0.0004649241012312578,
"weight_decay": 3.7624434569437365e-05,
"backbone_lr_ratio": 0.17310516991766065,
"moe_n_experts": 4,
"moe_top_k": 2,
"moe_expert_hidden_mult": 1,
"llm_lora_r": 8,
"d_model": 256,
"num_heads": 8,
"n_attn_layers": 4,
"fusion_strategy": "attention",
"head_hidden_dim": 128,
"set_transformer_block": "sab"
}

1
models/abl/both_biot5/20260624_015624/outer_fold_4/epoch_mean.json Normal file
View File

@ -0,0 +1 @@
{"epoch_mean": 16}

374
models/abl/both_biot5/20260624_015624/outer_fold_4/history.json Normal file
View File

@ -0,0 +1,374 @@
{
"train": [
{
"loss": 5.84952301979065,
"loss_size": 1.1317467346787453,
"loss_pdi": 0.7143913447856903,
"loss_ee": 1.098674190044403,
"loss_delivery": 1.0548573106527328,
"loss_biodist": 1.1708914518356324,
"loss_toxic": 0.6616318464279175,
"loss_moe_lb": 2.2185822248458864
},
{
"loss": 5.325460577011109,
"loss_size": 1.123024109005928,
"loss_pdi": 0.6624909400939941,
"loss_ee": 1.062855052947998,
"loss_delivery": 0.9382892578840256,
"loss_biodist": 0.9555662930011749,
"loss_toxic": 0.5752664297819138,
"loss_moe_lb": 2.040507411956787
},
{
"loss": 4.903134703636169,
"loss_size": 1.0160009294748307,
"loss_pdi": 0.6610889256000518,
"loss_ee": 1.0071192681789398,
"loss_delivery": 0.9504677027463913,
"loss_biodist": 0.7871204078197479,
"loss_toxic": 0.47663646638393403,
"loss_moe_lb": 2.069359838962555
},
{
"loss": 4.4732402801513675,
"loss_size": 1.0268311202526093,
"loss_pdi": 0.647139310836792,
"loss_ee": 0.991056752204895,
"loss_delivery": 0.8138237982988358,
"loss_biodist": 0.6108177661895752,
"loss_toxic": 0.37951129525899885,
"loss_moe_lb": 2.067969536781311
},
{
"loss": 4.059791779518127,
"loss_size": 1.0210174188017844,
"loss_pdi": 0.6079375505447387,
"loss_ee": 0.9449227213859558,
"loss_delivery": 0.7743113249540329,
"loss_biodist": 0.48670739829540255,
"loss_toxic": 0.21795016154646873,
"loss_moe_lb": 2.1050040245056154
},
{
"loss": 3.7733126401901247,
"loss_size": 0.9156547322869301,
"loss_pdi": 0.564486613869667,
"loss_ee": 0.9272694110870361,
"loss_delivery": 0.7476003125309945,
"loss_biodist": 0.4370688647031784,
"loss_toxic": 0.1732882007956505,
"loss_moe_lb": 2.1150545120239257
},
{
"loss": 3.6928714513778687,
"loss_size": 0.9506166368722916,
"loss_pdi": 0.582810002565384,
"loss_ee": 0.8964648246765137,
"loss_delivery": 0.7319391250610352,
"loss_biodist": 0.4102839112281799,
"loss_toxic": 0.11440396141260863,
"loss_moe_lb": 2.115168237686157
},
{
"loss": 3.5076019287109377,
"loss_size": 0.8235089927911758,
"loss_pdi": 0.5765671640634537,
"loss_ee": 0.8633758723735809,
"loss_delivery": 0.6743318244814873,
"loss_biodist": 0.3969142347574234,
"loss_toxic": 0.1655491843819618,
"loss_moe_lb": 2.1210803747177125
},
{
"loss": 3.4865591287612916,
"loss_size": 0.889287582039833,
"loss_pdi": 0.5686083287000656,
"loss_ee": 0.8519794940948486,
"loss_delivery": 0.6947177365422249,
"loss_biodist": 0.3592690348625183,
"loss_toxic": 0.11710096411406994,
"loss_moe_lb": 2.1221842765808105
},
{
"loss": 3.239002156257629,
"loss_size": 0.7932117432355881,
"loss_pdi": 0.5386495292186737,
"loss_ee": 0.8172129392623901,
"loss_delivery": 0.6377535864710808,
"loss_biodist": 0.3410655722022057,
"loss_toxic": 0.10361879635602236,
"loss_moe_lb": 2.097758674621582
},
{
"loss": 3.1010780334472656,
"loss_size": 0.7440196856856346,
"loss_pdi": 0.51452856361866,
"loss_ee": 0.7996897518634796,
"loss_delivery": 0.5864934250712395,
"loss_biodist": 0.33405207842588425,
"loss_toxic": 0.11317417323589325,
"loss_moe_lb": 2.1336498975753786
},
{
"loss": 3.569478893280029,
"loss_size": 1.1262501627206802,
"loss_pdi": 0.5549758553504944,
"loss_ee": 0.8099600613117218,
"loss_delivery": 0.6488962709903717,
"loss_biodist": 0.31612507402896883,
"loss_toxic": 0.1109379168599844,
"loss_moe_lb": 2.1338989973068236
},
{
"loss": 3.067129898071289,
"loss_size": 0.7431168645620346,
"loss_pdi": 0.5138901233673095,
"loss_ee": 0.7773933708667755,
"loss_delivery": 0.6177817612886429,
"loss_biodist": 0.3151536762714386,
"loss_toxic": 0.09142515510320663,
"loss_moe_lb": 2.1839953899383544
},
{
"loss": 3.4763768196105955,
"loss_size": 1.140934093296528,
"loss_pdi": 0.5292034894227982,
"loss_ee": 0.7735048115253449,
"loss_delivery": 0.6211386770009995,
"loss_biodist": 0.3026252746582031,
"loss_toxic": 0.10576980039477349,
"loss_moe_lb": 2.1303544521331785
},
{
"loss": 3.1925764083862305,
"loss_size": 0.7406455308198929,
"loss_pdi": 0.5184139549732208,
"loss_ee": 0.8398787379264832,
"loss_delivery": 0.6799886286258697,
"loss_biodist": 0.31126547902822493,
"loss_toxic": 0.09706502184271812,
"loss_moe_lb": 2.1912262439727783
},
{
"loss": 3.0583478689193724,
"loss_size": 0.7390100821852684,
"loss_pdi": 0.5151003897190094,
"loss_ee": 0.7950734734535218,
"loss_delivery": 0.5999791741371154,
"loss_biodist": 0.3222853854298592,
"loss_toxic": 0.07979281768202781,
"loss_moe_lb": 2.160543990135193
}
],
"val": [
{
"loss": 117.92023849487305,
"loss_size": 34.51820373535156,
"loss_pdi": 1.140266131842509,
"loss_ee": 6.3848408460617065,
"loss_delivery": 55.96383285522461,
"loss_biodist": 16.008755207061768,
"loss_toxic": 4.283290028572075,
"loss_moe_lb": 4.000000238418579,
"acc_pdi": 0.30303030303030304,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.9444444444444444
},
{
"loss": 53.49994659423828,
"loss_size": 17.449716567993164,
"loss_pdi": 0.5317580029368401,
"loss_ee": 3.8416613340377808,
"loss_delivery": 18.21935224533081,
"loss_biodist": 10.365546703338623,
"loss_toxic": 3.38681381939773,
"loss_moe_lb": 3.9999998807907104,
"acc_pdi": 0.30303030303030304,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.9444444444444444
},
{
"loss": 20.55763816833496,
"loss_size": 6.215997815132141,
"loss_pdi": 0.6493507921695709,
"loss_ee": 1.9068533182144165,
"loss_delivery": 4.330347537994385,
"loss_biodist": 5.752239227294922,
"loss_toxic": 1.832490186963696,
"loss_moe_lb": 3.987788677215576,
"acc_pdi": 0.30303030303030304,
"acc_ee": 0.45454545454545453,
"acc_toxic": 0.9444444444444444
},
{
"loss": 6.42457127571106,
"loss_size": 1.6664372086524963,
"loss_pdi": 0.6485741138458252,
"loss_ee": 0.9914509952068329,
"loss_delivery": 0.913701394572854,
"loss_biodist": 1.6495152115821838,
"loss_toxic": 0.5655998587608337,
"loss_moe_lb": 2.9400546550750732,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.5151515151515151,
"acc_toxic": 0.9444444444444444
},
{
"loss": 4.4594786167144775,
"loss_size": 0.5534357726573944,
"loss_pdi": 0.4115113094449043,
"loss_ee": 0.7405132353305817,
"loss_delivery": 0.7381824105978012,
"loss_biodist": 0.3299841731786728,
"loss_toxic": 1.667790524661541,
"loss_moe_lb": 2.9193990230560303,
"acc_pdi": 0.6666666666666666,
"acc_ee": 0.5151515151515151,
"acc_toxic": 0.8888888888888888
},
{
"loss": 4.24625551700592,
"loss_size": 0.47399845719337463,
"loss_pdi": 0.37447335571050644,
"loss_ee": 0.6216379702091217,
"loss_delivery": 0.8225250989198685,
"loss_biodist": 0.23464591801166534,
"loss_toxic": 1.695876032114029,
"loss_moe_lb": 2.7073097229003906,
"acc_pdi": 0.696969696969697,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.700586199760437,
"loss_size": 0.4298875629901886,
"loss_pdi": 0.3526789993047714,
"loss_ee": 0.5631953850388527,
"loss_delivery": 0.7401821315288544,
"loss_biodist": 0.18859301507472992,
"loss_toxic": 1.3968672528862953,
"loss_moe_lb": 2.5385050773620605,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.3784053325653076,
"loss_size": 0.464298814535141,
"loss_pdi": 0.33684471249580383,
"loss_ee": 0.5343593508005142,
"loss_delivery": 0.7194234766066074,
"loss_biodist": 0.17165696993470192,
"loss_toxic": 1.1205035001039505,
"loss_moe_lb": 2.4660756587982178,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.4850120544433594,
"loss_size": 0.45032067596912384,
"loss_pdi": 0.3273533768951893,
"loss_ee": 0.5187393873929977,
"loss_delivery": 0.7555239461362362,
"loss_biodist": 0.1526273936033249,
"loss_toxic": 1.2468775287270546,
"loss_moe_lb": 2.437071681022644,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.476802945137024,
"loss_size": 0.48352840542793274,
"loss_pdi": 0.32905983179807663,
"loss_ee": 0.5185238048434258,
"loss_delivery": 0.7266391441226006,
"loss_biodist": 0.1421719416975975,
"loss_toxic": 1.2420450150966644,
"loss_moe_lb": 2.401020050048828,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.5909444093704224,
"loss_size": 0.48948831856250763,
"loss_pdi": 0.3384985066950321,
"loss_ee": 0.5149989016354084,
"loss_delivery": 0.7514900229871273,
"loss_biodist": 0.13358022831380367,
"loss_toxic": 1.3266318514943123,
"loss_moe_lb": 2.4019196033477783,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.3500335216522217,
"loss_size": 0.4784044027328491,
"loss_pdi": 0.3424891047179699,
"loss_ee": 0.512165080755949,
"loss_delivery": 0.7664889544248581,
"loss_biodist": 0.12627345137298107,
"loss_toxic": 1.0899561867117882,
"loss_moe_lb": 2.3705557584762573,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.3171006441116333,
"loss_size": 0.48589515686035156,
"loss_pdi": 0.34011128172278404,
"loss_ee": 0.508001621812582,
"loss_delivery": 0.7478719539940357,
"loss_biodist": 0.12562820129096508,
"loss_toxic": 1.0740849822759628,
"loss_moe_lb": 2.439358711242676,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.3243151903152466,
"loss_size": 0.48700064420700073,
"loss_pdi": 0.34402309358119965,
"loss_ee": 0.5066678039729595,
"loss_delivery": 0.7546937242150307,
"loss_biodist": 0.12385652400553226,
"loss_toxic": 1.0728570520877838,
"loss_moe_lb": 2.4065319299697876,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.394656181335449,
"loss_size": 0.5005897283554077,
"loss_pdi": 0.3440004512667656,
"loss_ee": 0.5094963945448399,
"loss_delivery": 0.760299552232027,
"loss_biodist": 0.12519437819719315,
"loss_toxic": 1.119776889681816,
"loss_moe_lb": 2.382230758666992,
"acc_pdi": 0.7272727272727273,
"acc_ee": 0.6060606060606061,
"acc_toxic": 0.8888888888888888
},
{
"loss": 3.334527611732483,
"loss_size": 0.47870394587516785,
"loss_pdi": 0.3390600197017193,
"loss_ee": 0.5079137235879898,
"loss_delivery": 0.7363011240959167,
"loss_biodist": 0.12427961453795433,
"loss_toxic": 1.1121853068470955,
"loss_moe_lb": 2.389655351638794,
"acc_pdi": 0.7575757575757576,
"acc_ee": 0.6363636363636364,
"acc_toxic": 0.8888888888888888
}
]
}

Some files were not shown because too many files have changed in this diff Show More