gaofujie1997/ECG-ADGAN

ECG-ADGAN

Introduction

This repository is for the paper "A Novel Generative Adversarial Network based Electrocardiography Anomaly Detection".

Description

data_process.py

process raw data to *.pkl

train.py

train model

metric.py

evaluation model

Requirement

matplotlib==3.1.3
PyWavelets==1.1.1
wfdb==3.4.0
numpy==1.18.5
scikit_learn==1.0.2
tensorflow==2.7.0

Dataset

MIT-BIH Arrhythmia Database: https://physionet.org/content/mitdb/1.0.0/