ecgmentations 0.0.4

Ecg augmentation library and easy to use wrapper around other libraries

Ecgmentations

Ecgmentations is a Python library for ecg augmentation. Ecg augmentation is used in deep learning to increase the quality of trained models. The purpose of ecg augmentation is to create new training samples from the existing data.

Table of contents

• Authors
• Installation
• A simple example
• Citing

Authors

Rostislav Epifanov — Researcher at Novosibirsk State University

Installation

Installation from PyPI:

pip install ecgmentations

Installation from GitHub:

pip install git+https://github.com/rostepifanov/ecgmentations

A simple example

import numpy as np
import ecgmentations as E

# Declare an augmentation pipeline
transform = E.Sequential([
    E.Reverse(p=0.5),
    E.ChannelShuffle(p=0.06),
])

# Create example ecg
ecg = np.ones((12, 5000)).T

# Augment an ecg
transformed = transform(ecg=ecg)
transformed_ecg = transformed['ecg']

Citing

If you find this library useful for your research, please consider citing:

@misc{epifanov2023ecgmentations,
  Author = {Rostislav Epifanov},
  Title = {Ecgmentations},
  Year = {2023},
  Publisher = {GitHub},
  Journal = {GitHub repository},
  Howpublished = {\url{https://github.com/rostepifanov/ecgmentations}}
}