You can deploy this model and start earning money today!

hellothisisaswin/
ecg-ptbxl-classification

Name: ecg-ptbxl-classification
Rating: 2.5 (126 reviews)

keras

ecg

cardiovascular

multi-label-classification

ptb-xl

mit

Views

No views yet

ECG Cardiovascular Disease Classification

Multi-label classification of 5 cardiovascular superclasses (NORM, MI, STTC, CD, HYP) from 12-lead ECG recordings, trained on PTB-XL.

Deployed model: CNN (noaug training variant)

Files

ecg_model.keras | trained model
normalisation_params.npz | per-channel mean and std (z-score, from training fold)
thresholds.json | per-class decision thresholds optimised on the validation fold

Usage

python

import keras, numpy as np, json
from huggingface_hub import hf_hub_download

model = keras.saving.load_model(
    hf_hub_download("Steenslid/ecg-ptbxl-classification", "ecg_model.keras"))
params = np.load(hf_hub_download("Steenslid/ecg-ptbxl-classification", "normalisation_params.npz"))
with open(hf_hub_download("Steenslid/ecg-ptbxl-classification", "thresholds.json")) as f:
    thresholds = json.load(f)

# Input x: (1000, 12) float32 ECG in mV, 100 Hz, standard 12-lead order
x_norm = (x - params["mean"]) / params["std"]
probs = model.predict(x_norm[np.newaxis])[0]
preds = {sc: probs[i] >= thresholds[sc] for i, sc in enumerate(
    ["NORM","MI","STTC","CD","HYP"])}

Authors: Edvard Vindenes Steenslid & Morten Kvamme

hellothisisaswin/ecg-ptbxl-classification

ECG Cardiovascular Disease Classification

Files

Usage

hellothisisaswin/
ecg-ptbxl-classification