ECG heartbeat classification

Pytorch implementation for ECG heartbeat classification using Generative approaches.

link

• Python 3.3+
• yPtorch
• Add more...

First, download dataset with:

$ python ....

To train a model with downloaded dataset:

$ python ...
$ python ...

To test with an existing model:

$ python main.py ...
$ python main.py ...

After 6th epoch:

After 10th epoch:

MNIST codes are written by @PhoenixDai.

More results can be found here and here.

Details of the loss of Discriminator and Generator.

Details of the histogram of true and fake result of discriminator.

• ...

Tomer Golany