Nianfei commited on
Commit
cb36358
·
verified ·
1 Parent(s): 179c8da

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. README.md +67 -0
  2. hparams.yaml +13 -0
  3. model_weights.pth +3 -0
  4. training_info.json +17 -0
README.md ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: pytorch
3
+ tags:
4
+ - biosignals
5
+ - testmodel
6
+ metrics:
7
+ - validation_loss
8
+ ---
9
+ # Model Card for Testmodel
10
+
11
+
12
+ - Collection: NeuralLib: Deep Learning Models for Biosignals Processing
13
+
14
+ - Description: This is a Test
15
+
16
+
17
+ ```json
18
+ {
19
+ "architecture": "GRUseq2seq",
20
+ "model_name": "ECGPeakDetector",
21
+ "train_dataset": "private_gib01",
22
+ "biosignal": "ECG",
23
+ "sampling_frequency": 360,
24
+ "task": "peak detection",
25
+ "gpu_model": "NVIDIA GeForce GTX 1080 Ti",
26
+ "epochs": 80,
27
+ "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)",
28
+ "learning_rate": 0.001,
29
+ "validation_loss": 0.14879398047924042,
30
+ "training_time": 11375.492486476898,
31
+ "retraining": false,
32
+ "efficiency_flops": 0,
33
+ "efficiency_params": 0
34
+ }
35
+
36
+
37
+ ## Hyperparameters
38
+
39
+ bidirectional: true
40
+ dropout: 0
41
+ hid_dim:
42
+ - 32
43
+ - 64
44
+ - 64
45
+ learning_rate: 0.001
46
+ model_name: ECGPeakDetector
47
+ multi_label: true
48
+ n_features: 1
49
+ n_layers: 3
50
+ num_classes: 1
51
+ task: classification
52
+
53
+
54
+ # Example
55
+
56
+ import torch
57
+
58
+ from production_models import Testmodel
59
+
60
+ model = Testmodel()
61
+
62
+ signal = torch.rand(1, 100, 1) # Example input signal
63
+
64
+ predictions = model.predict(signal)
65
+
66
+ print(predictions)
67
+
hparams.yaml ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ bidirectional: true
2
+ dropout: 0
3
+ hid_dim:
4
+ - 32
5
+ - 64
6
+ - 64
7
+ learning_rate: 0.001
8
+ model_name: ECGPeakDetector
9
+ multi_label: true
10
+ n_features: 1
11
+ n_layers: 3
12
+ num_classes: 1
13
+ task: classification
model_weights.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67c5326c142ff01ec3e911f2459b42826392bf06dcc53ca2d26b051daaa785df
3
+ size 533628
training_info.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ }