Downloads · 30 days
7
19% of all-time downloads
ECG-Analysis-App
rajkalale/ECG-Analysis-App
ECG-Analysis-App is a machine learning model from rajkalale. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This is a lightweight CNN model trained for ECG image classification.
All-time downloads
37
Public
Repo size
27.3 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt27.3 MB · 100%
At a glance
- Model type
- simple_cnn
- Access
- Public
- Created
- Sep 19, 2025
- Updated
- Sep 19, 2025
- SHA
- d9397714
Model card
From the Hugging Face model README
ECG Analysis Model
This is a lightweight CNN model trained for ECG image classification.
Model Details
- Model Type: Convolutional Neural Network
- Classes: NORM, AFIB, SBRAD, STACH, BIGEMINY, VT
- Input Size: 224x224x3 (RGB images)
- Framework: PyTorch
Usage
import torch
from model import SimpleCNN
model = SimpleCNN(num_classes=6)
model.load_state_dict(torch.load('model.pt'))
model.eval()
Training Data
- Synthetic ECG patterns
- 6 classes: NORM, AFIB, SBRAD, STACH, BIGEMINY, VT
- Training samples: 800
- Test samples: 200
Created: 2025-09-19 15:23:09
- Type
- simple_cnn
- Created
- Sep 19, 2025
- Updated
- Sep 19, 2025