Skip to content

DeepECG-Kit

A deep learning library for ECG analysis and arrhythmia classification.

DeepECG-Kit provides 21 model architectures, 5 ECG datasets, and a complete training/evaluation pipeline — all without framework dependencies beyond PyTorch.

Quick Install

pip install deepecgkit

Quick Start

from deepecgkit.datasets import ECGDataModule
from deepecgkit.registry import get_dataset, get_dataset_info, get_model
from deepecgkit.training import ECGTrainer

dataset_info = get_dataset_info("af-classification")
dataset_class = get_dataset("af-classification")
data_module = ECGDataModule(dataset_class=dataset_class, batch_size=32, download=True)
data_module.setup(stage="fit")

model_class = get_model("kanres")
model = model_class(
    input_channels=dataset_info["input_channels"],
    output_size=dataset_info["num_classes"],
)

trainer = ECGTrainer(
    model=model,
    train_config={"learning_rate": 1e-3, "scheduler": {"factor": 0.5, "patience": 5}},
)
trainer.fit(data_module, epochs=50)
trainer.test(data_module)

Or use the CLI:

deepecg train -m kanres -d af-classification --epochs 50

Features

  • 21 model architectures — CNNs, ResNets, RNNs, Transformers, Mamba, and hybrids
  • 5 ECG datasets — PhysioNet 2017, PTB-XL, MIT-BIH AFDB, LTAFDB, and unified AF
  • Pure PyTorch — No framework dependencies beyond PyTorch itself
  • CLI & Python API — Train, evaluate, and predict from the command line or code
  • Registry system — Decorator-based model/dataset registration for extensibility
  • Pretrained weights — Available for select models (KanRes, AFModel)

Next Steps