JakJak504's picture
Update README.md
1b542da verified
metadata
license: apache-2.0
language:
  - en

DeeECG EfficientNet V2 Fine Tuned to detect AFIB at 5y

  • model_name: efficientnetv2_afib_5y
  • Utilizes the EfficientNetV2 architecture for processing ECG signals.
  • Pre-trained model is loaded from a specified directory, and the model is used to process ECG signal tensors.
  • The model takes an input signal tensor and outputs the logits for the binary classification of incident AFIB at 5 years.
  • Model compiled with JIT,

To load use

import torch
model = torch.jit.load(model_path)