README.md

metadata

library_name: pytorch
tags:
  - biosignals
  - testmodel
metrics:
  - validation_loss

Model Card for Testmodel

• Collection: NeuralLib: Deep Learning Models for Biosignals Processing

• Description: This is a Test

{
    "architecture": "GRUseq2seq",
    "model_name": "ECGPeakDetector",
    "train_dataset": "private_gib01",
    "biosignal": "ECG",
    "sampling_frequency": 360,
    "task": "peak detection",
    "gpu_model": "NVIDIA GeForce GTX 1080 Ti",
    "epochs": 80,
    "optimizer": "Adam (\nParameter Group 0\n    amsgrad: False\n    betas: (0.9, 0.999)\n    capturable: False\n    differentiable: False\n    eps: 1e-08\n    foreach: None\n    fused: None\n    initial_lr: 0.001\n    lr: 0.001\n    maximize: False\n    weight_decay: 1e-05\n)",
    "learning_rate": 0.001,
    "validation_loss": 0.14879398047924042,
    "training_time": 11375.492486476898,
    "retraining": false,
    "efficiency_flops": 0,
    "efficiency_params": 0
}


## Hyperparameters

bidirectional: true
dropout: 0
hid_dim:
- 32
- 64
- 64
learning_rate: 0.001
model_name: ECGPeakDetector
multi_label: true
n_features: 1
n_layers: 3
num_classes: 1
task: classification


# Example

import torch

from production_models import Testmodel

model = Testmodel()

signal = torch.rand(1, 100, 1)  # Example input signal

predictions = model.predict(signal)

print(predictions)