# 3-Class Beat Classification (BEAT-3-EFF-SM)

## Overview

The following table provides the latest pre-trained model for 3-class beat classification. Below we also provide additional details including training configuration, performance metrics, and downloads.

| NAME            | DATASET    | FS    | DURATION | # CLASSES | MODEL          | PARAMS | FLOPS   | METRIC   |
| --------------- | ---------- | ----- | -------- | --------- | -------------- | ------ | ------- | -------- |
| __BC-3-EFF-SM__ | Icentia11k | 100Hz | 5s       | 3         | EfficientNetV2 | 41K    | 2.1M    | 92.0% F1 |

---

## Input

The model is trained on 5-second, raw ECG frames sampled at 100 Hz.

- **Sensor**: ECG
- **Location**: Wrist
- **Sampling Rate**: 100 Hz
- **Frame Size**: 5 seconds

---

## Class Mapping

The model is able to distinguish between normal sinus rhythm (NSR), premature/ectopic atrial contractions (PAC), and premature/ectopic ventricular contractions (PVC). The class mapping is as follows:

| Base Class    | Target Class | Label     |
| ------------- | ------------ | --------- |
| 0-NSR         | 0            | NSR       |
| 1-PAC         | 1            | PAC       |
| 2-PVC         | 2            | PVC       |

---

## Dataset

The model is trained on the following datasets:

- **[Icentia11k](https://ambiqai.github.io/heartkit/datasets/icentia11k/)**

---

## Model Performance

The confusion matrix for the model is depicted below.

---

## Downloads

| Asset                                                                | Description                   |
| -------------------------------------------------------------------- | ----------------------------- |
| [configuration.json](https://ambiqai-model-zoo.s3.us-west-2.amazonaws.com/heartkit/beat/beat-3-eff-sm/latest/configuration.json)   | Configuration file            |
| [model.keras](https://ambiqai-model-zoo.s3.us-west-2.amazonaws.com/heartkit/beat/beat-3-eff-sm/latest/model.keras)            | Keras Model file              |
| [model.tflite](https://ambiqai-model-zoo.s3.us-west-2.amazonaws.com/heartkit/beat/beat-3-eff-sm/latest/model.tflite)       | TFLite Model file             |
| [metrics.json](https://ambiqai-model-zoo.s3.us-west-2.amazonaws.com/heartkit/beat/beat-3-eff-sm/latest/metrics.json)       | Metrics file                  |

---