File size: 666 Bytes
5a218d0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
{
"model_name": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract",
"task": "binary_classification",
"labels": {
"0": "Non-OHCA",
"1": "OHCA"
},
"max_length": 512,
"optimal_threshold": 0.12000000000000001,
"test_metrics": {
"sensitivity": 0.8,
"specificity": 0.95,
"precision": 0.8,
"npv": 0.95,
"f1": 0.8000000000000002,
"auc": 0.905,
"threshold": 0.12000000000000001
},
"training_params": {
"epochs": 5,
"batch_size": 4,
"learning_rate": 2e-05,
"weight_decay": 0.01,
"sections_used": [
"chief_complaint",
"hpi",
"ems",
"ed_course",
"impression"
]
}
} |