JakJak504's picture
Update README.md
f3d8d94 verified
metadata
license: apache-2.0

DeepECG EfficientNet V2 Fine Tuned to detect LVEF under 50 % model

  • model_name: efficientnetv2_lvef_under_50
  • 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 < 50%.
  • Model compiled with JIT,
  • Frozen weights no tunable

To load use

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