This is a CNN based model which aims to automatically classify the ECG signals of a normal patient vs. a patient with AF and has been trained to achieve up to 93.33% validation accuracy.
A CNN based to classify the ECG signals of a patient with and without Atrial Fibrillation. Accuracy Achieved: 93.33%.
Data Preprocessing Notebook
·
Model Notebook
·
Report Bug
This is a CNN based model which aims to automatically classify the ECG signals of a normal patient vs. a patient with Atrial Fibrillation and has been trained to achieve up to 93.33%
validation accuracy.
The CNN used here is 1D Convolutional Neural Networks.
Note: The dataset has been moved from the original URL. Hence, the dataset provided above may be different from the one that was originally used to create this project.
The experiment should be fairly reproducible. However, a GPU would be recommended for training. For Inference, a CPU System would suffice.
Data_Preprocessing.ipynb
to be able to read the signals from the downloaded CSVs.Model.ipynb
read the saved processed dataset and train the model.model.predict()
function after the model
has been trained successfully.Alternative Option: Google Colaboratory - GPU Kernel
Simple List of Deep Learning Libraries. The main Architecture/Model is developed with Keras, which comes as a part of Tensorflow 1.x
Since this is a Proof of Concept Project, I am not maintaining a CHANGELOG.md at the moment. However, the primary goal is to improve the architecture to make the predicted masks more accurate.
Contributions are what make the open source community such an amazing place to be learn, inspire, and create. Any contributions you make are greatly appreciated.
git checkout -b feature/AmazingFeature
)git commit -m 'Add some AmazingFeature'
)git push origin feature/AmazingFeature
)Distributed under the MIT License. See LICENSE for more information.
This is a preliminary attempt and is a work-in-progress: