hubii-world/ecg-to-hrv-pipeline

General

This repository contains two preprocessing pipelines: ECG2HRV and Rpeaks2HRV

ECG2HRVPipeline

Pipeline for processing raw ECG signals towards HRV features. For more details see HUBII

How to use

To use the model, you will need the transformers library of HuggingFace:

# Imports
from transformers import pipeline

ecg2hrv_pipeline = pipeline(model = "hubii-world/ecg-to-hrv-pipeline", trust_remote_code=True)

Example usage of the model:

# ecg should be a 1D numpy array with the ECG signal
hrv_features = ecg2hrv_pipeline(ecg)
# returns hrv_features in a dictionary with the feature names as keys

RPeaks2HRVPipeline

Less powerful version of the ECG2HRVPipeline. Processes identified peaks towards HRV features.

How to use

Similar to the usage of ECG2HRVPipeline, simply fetch a different model name:

# Imports
from transformers import pipeline

rpeaks2hrv_pipeline = pipeline(model = "hubii-world/rpeaks-to-hrv-pipeline", trust_remote_code=True)

Example usage of the model:

# ecg shape: TODO
hrv_features = rpeaks2hrv_pipeline(peaks)
# returns hrv_features in a dictionary with the feature names as keys