lnp_ml/models/abl_full/rf/seed42/summary.json

602 lines
14 KiB
JSON

{
"fold_results": [
{
"fold": 0,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.3484356700772522,
"rmse": 1.1612216283196124,
"mae": 0.4798589242061937,
"r2": 0.1747241783006852
},
"delivery": {
"n_samples": 58,
"mse": 0.3976989005700695,
"rmse": 0.6306337293311146,
"mae": 0.4073375633664443,
"r2": 0.49450069866587976
},
"toxic": {
"n_samples": 58,
"accuracy": 0.9655172413793104,
"precision": 0.75,
"recall": 0.9821428571428572,
"f1": 0.8242424242424242,
"true_class_counts": [
56,
2
],
"pred_class_counts": [
54,
4
],
"confusion_matrix": [
[
54,
2
],
[
0,
2
]
]
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7857142857142857,
"precision": 0.7551020408163265,
"recall": 0.8118317890235209,
"f1": 0.7630094043887148,
"true_class_counts": [
61,
23
],
"pred_class_counts": [
49,
35
],
"confusion_matrix": [
[
46,
15
],
[
3,
20
]
]
},
"ee": {
"n_samples": 84,
"accuracy": 0.75,
"precision": 0.7092916445857623,
"recall": 0.7614285714285715,
"f1": 0.7175824990730441,
"true_class_counts": [
14,
20,
50
],
"pred_class_counts": [
17,
30,
37
],
"confusion_matrix": [
[
10,
3,
1
],
[
3,
17,
0
],
[
4,
10,
36
]
]
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.15998570647485624,
"js_divergence": 0.035472610690609384
}
}
},
{
"fold": 1,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.30493822479834903,
"rmse": 0.5522121193874225,
"mae": 0.34891908283267253,
"r2": 0.06882707391945686
},
"delivery": {
"n_samples": 61,
"mse": 0.7752937492958416,
"rmse": 0.8805076656655759,
"mae": 0.5151286462071908,
"r2": 0.4131585218812348
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9508196721311475,
"precision": 0.75,
"recall": 0.9741379310344828,
"f1": 0.8200589970501475,
"true_class_counts": [
58,
3
],
"pred_class_counts": [
55,
6
],
"confusion_matrix": [
[
55,
3
],
[
0,
3
]
]
},
"pdi": {
"n_samples": 84,
"accuracy": 0.6547619047619048,
"precision": 0.5803571428571429,
"recall": 0.5952380952380952,
"f1": 0.5822328931572629,
"true_class_counts": [
63,
21
],
"pred_class_counts": [
56,
28
],
"confusion_matrix": [
[
45,
18
],
[
11,
10
]
]
},
"ee": {
"n_samples": 84,
"accuracy": 0.5476190476190477,
"precision": 0.5001165151969107,
"recall": 0.4989534275248561,
"f1": 0.48349530196256724,
"true_class_counts": [
13,
22,
49
],
"pred_class_counts": [
19,
31,
34
],
"confusion_matrix": [
[
5,
7,
1
],
[
8,
11,
3
],
[
6,
13,
30
]
]
},
"biodist": {
"n_samples": 61,
"kl_divergence": 0.1177531067102451,
"js_divergence": 0.027774748827162944
}
}
},
{
"fold": 2,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.3091245824692921,
"rmse": 0.5559897323416073,
"mae": 0.4046506446335209,
"r2": 0.3894844727418765
},
"delivery": {
"n_samples": 60,
"mse": 0.623471858388545,
"rmse": 0.789602341934562,
"mae": 0.4688382081680016,
"r2": 0.17005089391015027
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9344262295081968,
"precision": 0.7142857142857143,
"recall": 0.9655172413793103,
"f1": 0.7821428571428571,
"true_class_counts": [
58,
3
],
"pred_class_counts": [
54,
7
],
"confusion_matrix": [
[
54,
4
],
[
0,
3
]
]
},
"pdi": {
"n_samples": 84,
"accuracy": 0.6904761904761905,
"precision": 0.6693363844393593,
"recall": 0.717008797653959,
"f1": 0.662962962962963,
"true_class_counts": [
62,
22
],
"pred_class_counts": [
46,
38
],
"confusion_matrix": [
[
41,
21
],
[
5,
17
]
]
},
"ee": {
"n_samples": 84,
"accuracy": 0.6428571428571429,
"precision": 0.5842305037957212,
"recall": 0.6349206349206349,
"f1": 0.5925185925185925,
"true_class_counts": [
12,
21,
51
],
"pred_class_counts": [
21,
23,
40
],
"confusion_matrix": [
[
8,
2,
2
],
[
5,
12,
4
],
[
8,
9,
34
]
]
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.11799692057231032,
"js_divergence": 0.029125166530752162
}
}
},
{
"fold": 3,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.6732277890580922,
"rmse": 1.2935330645399414,
"mae": 0.5211672051452704,
"r2": 0.05360960932568781
},
"delivery": {
"n_samples": 59,
"mse": 0.8527625246024666,
"rmse": 0.9234514197306032,
"mae": 0.5525073854104011,
"r2": 0.17101350859831832
},
"toxic": {
"n_samples": 60,
"accuracy": 0.9333333333333333,
"precision": 0.7142857142857143,
"recall": 0.9649122807017544,
"f1": 0.7818181818181817,
"true_class_counts": [
57,
3
],
"pred_class_counts": [
53,
7
],
"confusion_matrix": [
[
53,
4
],
[
0,
3
]
]
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7380952380952381,
"precision": 0.6612903225806452,
"recall": 0.6612903225806452,
"f1": 0.6612903225806452,
"true_class_counts": [
62,
22
],
"pred_class_counts": [
62,
22
],
"confusion_matrix": [
[
51,
11
],
[
11,
11
]
]
},
"ee": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.5872039589961117,
"recall": 0.6381096028154852,
"f1": 0.5938808373590982,
"true_class_counts": [
13,
20,
51
],
"pred_class_counts": [
23,
20,
41
],
"confusion_matrix": [
[
10,
3,
0
],
[
9,
8,
3
],
[
4,
9,
38
]
]
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.14886390502458902,
"js_divergence": 0.03503301624445018
}
}
},
{
"fold": 4,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.5439429374081691,
"rmse": 0.7375248723996833,
"mae": 0.41855190159962563,
"r2": 0.2721367833807379
},
"delivery": {
"n_samples": 58,
"mse": 0.8807361070482976,
"rmse": 0.9384754163260206,
"mae": 0.4923390269539427,
"r2": 0.14834272376574187
},
"toxic": {
"n_samples": 59,
"accuracy": 0.9661016949152542,
"precision": 0.8,
"recall": 0.9821428571428572,
"f1": 0.865909090909091,
"true_class_counts": [
56,
3
],
"pred_class_counts": [
54,
5
],
"confusion_matrix": [
[
54,
2
],
[
0,
3
]
]
},
"pdi": {
"n_samples": 84,
"accuracy": 0.6785714285714286,
"precision": 0.6428571428571428,
"recall": 0.6796187683284457,
"f1": 0.6415362731152205,
"true_class_counts": [
62,
22
],
"pred_class_counts": [
49,
35
],
"confusion_matrix": [
[
42,
20
],
[
7,
15
]
]
},
"ee": {
"n_samples": 84,
"accuracy": 0.6309523809523809,
"precision": 0.5673525820584645,
"recall": 0.6021367521367521,
"f1": 0.5701075744179193,
"true_class_counts": [
12,
20,
52
],
"pred_class_counts": [
17,
28,
39
],
"confusion_matrix": [
[
7,
4,
1
],
[
6,
11,
3
],
[
4,
13,
35
]
]
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.12072341813237816,
"js_divergence": 0.029660750070876794
}
}
}
],
"summary_stats": {
"size": {
"mse_mean": 0.8359338407622309,
"mse_std": 0.5671793909058501,
"rmse_mean": 0.8600962833976535,
"rmse_std": 0.3101100192637678,
"mae_mean": 0.43462955168345657,
"mae_std": 0.06006325295257579,
"r2_mean": 0.19175642353368885,
"r2_std": 0.12652560874612837
},
"delivery": {
"mse_mean": 0.705992627981044,
"mse_std": 0.1781763346919953,
"rmse_mean": 0.8325341145975752,
"rmse_std": 0.11348821970704938,
"mae_mean": 0.4872301660211962,
"mae_std": 0.048552308082167685,
"r2_mean": 0.279413269364265,
"r2_std": 0.14494172624677884
},
"toxic": {
"accuracy_mean": 0.9500396342534485,
"accuracy_std": 0.014290090778642328,
"precision_mean": 0.7457142857142858,
"precision_std": 0.03149343955006944,
"recall_mean": 0.9737706334802525,
"recall_std": 0.007575117962473199,
"f1_mean": 0.8148343102325404,
"f1_std": 0.031251364844790935
},
"pdi": {
"accuracy_mean": 0.7095238095238094,
"accuracy_std": 0.04677829215330594,
"precision_mean": 0.6617886067101234,
"precision_std": 0.05613216163532689,
"recall_mean": 0.6929975545649333,
"recall_std": 0.07132933119437145,
"f1_mean": 0.6622063712409613,
"f1_std": 0.05828494354685521
},
"ee": {
"accuracy_mean": 0.6476190476190476,
"accuracy_std": 0.0650309537321317,
"precision_mean": 0.5896390409265941,
"precision_std": 0.06762084933247267,
"recall_mean": 0.62710979776526,
"recall_std": 0.08398957802940274,
"f1_mean": 0.5915169610662443,
"f1_std": 0.07486808628755862
},
"biodist": {
"kl_divergence_mean": 0.13306461138287579,
"kl_divergence_std": 0.017822152356421223,
"js_divergence_mean": 0.031413258472770295,
"js_divergence_std": 0.003197698848519836
}
}
}