YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)
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
- Downloads last month
- -
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support