| { | |
| "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" | |
| ] | |
| } | |
| } |