Views
No views yet
1{
2 "architecture": "GRUseq2seq",
3 "model_name": "ECGPeakDetector",
4 "train_dataset": "private_gib01",
5 "biosignal": "ECG",
6 "sampling_frequency": 360,
7 "task": "peak detection",
8 "gpu_model": "NVIDIA GeForce GTX 1080 Ti",
9 "epochs": 80,
10 "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)",
11 "learning_rate": 0.001,
12 "validation_loss": 0.14879398047924042,
13 "training_time": 11375.492486476898,
14 "retraining": false,
15 "efficiency_flops": 0,
16 "efficiency_params": 0
17}
18
19
20## Hyperparameters
21
22bidirectional: true
23dropout: 0
24hid_dim:
25- 32
26- 64
27- 64
28learning_rate: 0.001
29model_name: ECGPeakDetector
30multi_label: true
31n_features: 1
32n_layers: 3
33num_classes: 1
34task: classification
35
36
37# Example
38
39import torch
40
41from production_models import Testmodel
42
43model = Testmodel()
44
45signal = torch.rand(1, 100, 1) # Example input signal
46
47predictions = model.predict(signal)
48
49print(predictions)
50