medresearch-subcop-multi-site-pool / experiment_metadata.json
codewithdark's picture
SubCoP: subgroup-conditional conformal prediction
fbf1f85 verified
Raw History Blame Contribute Delete
834 Bytes
{
"hyperparameters": {
"alpha": 0.1,
"epochs": 300,
"lr": 0.01,
"n_classes": 5,
"n_groups": 2,
"conformal_methods": [
"split",
"mondrian",
"aps",
"raps"
],
"pool_datasets": true
},
"final_metrics": {
"alpha": 0.1,
"nominal_coverage": 0.9,
"marginal_coverage": 0.9003202562049639,
"mondrian_coverage": 0.8955164131305045,
"avg_set_size_marginal": 1.97731575531584,
"avg_set_size_mondrian": 2.88834720296773,
"train_size": 3996,
"val_size": 999,
"calib_size": 2497,
"test_size": 2498,
"batch_size": 32,
"steps_per_epoch": 125,
"examples_seen": 1198800,
"global_step": 37375,
"early_stop_epoch": 31,
"early_stop_val_loss": 0.5498771071434021,
"min_calibration_per_group": 50,
"n_bootstrap_ci": 200
}
}