JakJak504's picture
Update README.md
973b356 verified
---
license: apache-2.0
language:
- en
---
# DeepECG EfficientNet V2 Fine Tuned to detect LVEF under or equal 40 % model
- model_name: `efficientnetv2_lvef_equal_under_40`
- 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 LVEF <= 40%.
- Model compiled with JIT,
- Frozen weights no tunable
To load use
```
import torch
model = torch.jit.load(model_path)
```