rajkalale commited on
Commit
d939771
·
verified ·
1 Parent(s): a016e31

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +28 -0
README.md ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ # ECG Analysis Model
3
+
4
+ This is a lightweight CNN model trained for ECG image classification.
5
+
6
+ ## Model Details
7
+ - **Model Type**: Convolutional Neural Network
8
+ - **Classes**: NORM, AFIB, SBRAD, STACH, BIGEMINY, VT
9
+ - **Input Size**: 224x224x3 (RGB images)
10
+ - **Framework**: PyTorch
11
+
12
+ ## Usage
13
+ ```python
14
+ import torch
15
+ from model import SimpleCNN
16
+
17
+ model = SimpleCNN(num_classes=6)
18
+ model.load_state_dict(torch.load('model.pt'))
19
+ model.eval()
20
+ ```
21
+
22
+ ## Training Data
23
+ - Synthetic ECG patterns
24
+ - 6 classes: NORM, AFIB, SBRAD, STACH, BIGEMINY, VT
25
+ - Training samples: 800
26
+ - Test samples: 200
27
+
28
+ Created: 2025-09-19 15:23:09