287 lines
8.1 KiB
JSON

{
"fold_results": [
{
"fold": 0,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.762732353807776,
"rmse": 1.3276793113578955,
"mae": 0.6979452131473156,
"r2": -0.07883558225954723
},
"delivery": {
"n_samples": 58,
"mse": 0.746449183391017,
"rmse": 0.8639729066301888,
"mae": 0.646006099558596,
"r2": 0.051218037285134965
},
"toxic": {
"n_samples": 58,
"accuracy": 0.9827586206896551,
"precision": 0.9912280701754386,
"recall": 0.75,
"f1": 0.8289085545722714
},
"pdi": {
"n_samples": 84,
"accuracy": 0.5119047619047619,
"precision": 0.4791666666666667,
"recall": 0.47434069850320737,
"f1": 0.46446897838594314
},
"ee": {
"n_samples": 84,
"accuracy": 0.7261904761904762,
"precision": 0.6766726403823178,
"recall": 0.6523809523809524,
"f1": 0.641367232698502
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.34443429112434387,
"js_divergence": 0.08364559710025787
}
}
},
{
"fold": 1,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.5181048119517085,
"rmse": 0.7197949791098216,
"mae": 0.5254934162256264,
"r2": -0.5821078937747561
},
"delivery": {
"n_samples": 61,
"mse": 1.1243467667528977,
"rmse": 1.0603521899599668,
"mae": 0.6867733584014607,
"r2": 0.14895057570408377
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9508196721311475,
"precision": 0.75,
"recall": 0.9741379310344828,
"f1": 0.8200589970501475
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7261904761904762,
"precision": 0.6071428571428571,
"recall": 0.5793650793650793,
"f1": 0.5849624060150376
},
"ee": {
"n_samples": 84,
"accuracy": 0.5952380952380952,
"precision": 0.5352564102564102,
"recall": 0.5345131059416773,
"f1": 0.5263532763532763
},
"biodist": {
"n_samples": 61,
"kl_divergence": 0.7807825803756714,
"js_divergence": 0.12687404453754425
}
}
},
{
"fold": 2,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.44417219953539205,
"rmse": 0.6664624517070651,
"mae": 0.5136315814618554,
"r2": 0.12276783858786378
},
"delivery": {
"n_samples": 60,
"mse": 0.6732209168800264,
"rmse": 0.8205004063862652,
"mae": 0.5596807344001717,
"r2": 0.10382628141775896
},
"toxic": {
"n_samples": 61,
"accuracy": 0.9180327868852459,
"precision": 0.6875,
"recall": 0.9568965517241379,
"f1": 0.7502047502047502
},
"pdi": {
"n_samples": 84,
"accuracy": 0.75,
"precision": 0.6652173913043478,
"recall": 0.625366568914956,
"f1": 0.636063544460491
},
"ee": {
"n_samples": 84,
"accuracy": 0.7142857142857143,
"precision": 0.6787755102040816,
"recall": 0.6622315592903828,
"f1": 0.6671936758893281
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.3705731928348541,
"js_divergence": 0.0945676639676094
}
}
},
{
"fold": 3,
"test_metrics": {
"size": {
"n_samples": 83,
"mse": 1.9476472096853905,
"rmse": 1.3955813160419535,
"mae": 0.6786922237201852,
"r2": -0.10160416549494755
},
"delivery": {
"n_samples": 59,
"mse": 1.0546120682688678,
"rmse": 1.0269430696337882,
"mae": 0.7278208352972643,
"r2": -0.025208277515615496
},
"toxic": {
"n_samples": 60,
"accuracy": 0.9166666666666666,
"precision": 0.6875,
"recall": 0.956140350877193,
"f1": 0.749791492910759
},
"pdi": {
"n_samples": 84,
"accuracy": 0.7738095238095238,
"precision": 0.7091007583965331,
"recall": 0.6414956011730205,
"f1": 0.6571428571428573
},
"ee": {
"n_samples": 84,
"accuracy": 0.6071428571428571,
"precision": 0.4344751866490997,
"recall": 0.443614881850176,
"f1": 0.4324842382393017
},
"biodist": {
"n_samples": 60,
"kl_divergence": 0.9395005106925964,
"js_divergence": 0.12415548413991928
}
}
},
{
"fold": 4,
"test_metrics": {
"size": {
"n_samples": 84,
"mse": 0.7430198511001034,
"rmse": 0.8619859924036488,
"mae": 0.5445426530481893,
"r2": 0.005747196126379084
},
"delivery": {
"n_samples": 58,
"mse": 1.0484304054163862,
"rmse": 1.0239289064268018,
"mae": 0.6575078364333202,
"r2": -0.013814952507488876
},
"toxic": {
"n_samples": 59,
"accuracy": 0.9661016949152542,
"precision": 0.8,
"recall": 0.9821428571428572,
"f1": 0.8659090909090909
},
"pdi": {
"n_samples": 84,
"accuracy": 0.6666666666666666,
"precision": 0.5142857142857142,
"recall": 0.5102639296187683,
"f1": 0.5050505050505051
},
"ee": {
"n_samples": 84,
"accuracy": 0.6547619047619048,
"precision": 0.6187822362094589,
"recall": 0.6354700854700855,
"f1": 0.6152751423149906
},
"biodist": {
"n_samples": 58,
"kl_divergence": 0.4118382930755615,
"js_divergence": 0.1084776222705841
}
}
}
],
"summary_stats": {
"size": {
"mse_mean": 1.083135285216074,
"mse_std": 0.6406941255063764,
"rmse_mean": 0.9943008101240769,
"rmse_std": 0.3074104490785543,
"mae_mean": 0.5920610175206343,
"mae_std": 0.0794441259199241,
"r2_mean": -0.1268065213630016,
"r2_std": 0.2408707365476069
},
"delivery": {
"mse_mean": 0.9294118681418391,
"mse_std": 0.18272890550921916,
"rmse_mean": 0.9591394958074021,
"rmse_std": 0.09727947226502257,
"mae_mean": 0.6555577728181625,
"mae_std": 0.05564855198139901,
"r2_mean": 0.052994332876774665,
"r2_std": 0.06689360454000906
},
"toxic": {
"accuracy_mean": 0.9468758882575938,
"accuracy_std": 0.02614298041760053,
"precision_mean": 0.7832456140350876,
"precision_std": 0.11222934338016616,
"recall_mean": 0.9238635381557341,
"recall_std": 0.08750485258067292,
"f1_mean": 0.8029745771294039,
"f1_std": 0.04590925112738489
},
"pdi": {
"accuracy_mean": 0.6857142857142857,
"accuracy_std": 0.09391944249198107,
"precision_mean": 0.5949826775592237,
"precision_std": 0.08721131763365055,
"recall_mean": 0.5661663755150064,
"recall_std": 0.06466900236745027,
"f1_mean": 0.5695376582109668,
"f1_std": 0.07421092597308579
},
"ee": {
"accuracy_mean": 0.6595238095238095,
"accuracy_std": 0.053558199424771406,
"precision_mean": 0.5887923967402736,
"precision_std": 0.09316564987347752,
"recall_mean": 0.5856421169866548,
"recall_std": 0.08436676188679884,
"f1_mean": 0.5765347130990797,
"f1_std": 0.08624597369019231
},
"biodist": {
"kl_divergence_mean": 0.5694257736206054,
"kl_divergence_std": 0.24356696856764282,
"js_divergence_mean": 0.10754408240318299,
"js_divergence_std": 0.016673227855747277
}
}
}