anastasiyayudo commited on
Commit
cbf80ea
·
verified ·
1 Parent(s): 76955f0

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +45 -3
README.md CHANGED
@@ -1,3 +1,45 @@
1
- ---
2
- license: unknown
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - tabular-classification
4
+ - healthcare
5
+ - synthetic-data
6
+ license: mit
7
+ ---
8
+
9
+ # HFrEF prediction model (synthetic EKG data)
10
+
11
+ XGBoost classifier that predicts **Heart Failure with Reduced Ejection Fraction (HFrEF)** from features derived from synthetic 12-lead EKG data and a synthetic lab panel.
12
+
13
+ **All training data is synthetic** - generated by our own generator. This model is a course/educational artifact and must NOT be used for any real clinical decision.
14
+
15
+ ## Model version
16
+
17
+ `model_20260912_135646.joblib`
18
+
19
+ ## Evaluation metrics
20
+
21
+ | Metric | Value |
22
+ |---|---|
23
+ | AUROC | 0.9720 |
24
+ | AUPRC | 0.6795 |
25
+ | Brier score | 0.0458 |
26
+ | ECE (calibration) | 0.0464 |
27
+
28
+ Confusion matrix (rows = true, cols = predicted):
29
+
30
+ ```
31
+ [[185, 2], [10, 3]]
32
+ ```
33
+
34
+ ## Important note on the label
35
+
36
+ The label (HFrEF: ejection fraction < 40) is deliberately NOT a feature. Ground truth arrives on a delay in a separate outcomes table, so the model never sees the answer at prediction time. This is what makes the AUROC honest (high but not perfect) and the monitoring meaningful.
37
+
38
+ ## How to load
39
+
40
+ ```python
41
+ from huggingface_hub import hf_hub_download
42
+ import joblib
43
+ path = hf_hub_download("anastasiyayudo/HFrEF", "model_20260912_135646.joblib")
44
+ model = joblib.load(path)
45
+ ```