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