SleepECG provides tools for sleep stage classification when EEG signals are not available. Based only on ECG, SleepECG provides functions for
- downloading and reading open polysomnography datasets,
- detecting heartbeats from ECG signals, and
- classifying sleep stages (which includes preprocessing, feature extraction, and classification).
Documentation for SleepECG is available on Read the Docs. Check out the changelog to learn what we added, changed, or fixed.
SleepECG is available on PyPI and can be installed with pip:
pip install sleepecg
Alternatively, an unofficial conda package is available:
conda install -c conda-forge sleepecg
SleepECG with all optional dependencies can be installed with the following command:
pip install "sleepecg[full]"
If you want the latest development version, use the following command:
pip install git+https://github.com/cbrnr/sleepecg
The following example detects heartbeats in a short ECG (a one-dimensional NumPy array):
import numpy as np
from sleepecg import detect_heartbeats, get_toy_ecg
ecg, fs = get_toy_ecg() # 5 min of ECG data at 360 Hz
beats = detect_heartbeats(ecg, fs) # indices of detected heartbeats
SleepECG requires Python ≥ 3.9 and the following packages:
Optional dependencies provide additional features:
- edfio ≥ 0.4.0 (read data from MESA and SHHS)
- joblib ≥ 1.0.0 (parallelized feature extraction)
- matplotlib ≥ 3.5.0 (plot ECG time courses, hypnograms, and confusion matrices)
- numba ≥ 0.59.1 (JIT-compiled heartbeat detector)
- tensorflow ≥ 2.16.1 (sleep stage classification with Keras models)
- wfdb ≥ 3.4.0 (read data from SLPDB, MITDB, and LTDB)
The contributing guide contains detailed instructions on how to contribute to SleepECG.