maude-clinicalbert / config_phase2.json
mukundisb's picture
Push fold-3 checkpoint (Death F1=0.8418, Weighted F1=0.8693)
9729d18 verified
Raw
History Blame Contribute Delete
616 Bytes
{
"pretrained_model": "emilyalsentzer/Bio_ClinicalBERT",
"pooling": "cls_mean_concat",
"max_length": 512,
"num_labels": 4,
"label_order": [
"D",
"I",
"M",
"O"
],
"dropout": 0.1,
"fold": 3,
"val_metrics": {
"f1_D": 0.8418,
"f1_I": 0.8749,
"f1_M": 0.9056,
"f1_O": 0.4764,
"f1_macro": 0.7747,
"f1_weighted": 0.8693
},
"cv_aggregated": {
"f1_D_mean": 0.8318,
"f1_D_std": 0.0076,
"f1_weighted_mean": 0.8691,
"f1_weighted_std": 0.001
},
"phase1_baseline": {
"f1_D": 0.7701,
"f1_weighted": 0.8484
}
}