Version 1.4.0.RC1

Author: apage224

configs/arr-2-eff-sm.json

@@ -36,6 +36,8 @@
 
     "val_size": 40000,
     "test_size": 40000,
+    "val_file": "val.tfds",
+    "test_file": "val.tfds",
 
     "val_patients": 0.20,
     "batch_size": 256,

configs/arr-4-eff-lg.json

@@ -11,6 +11,9 @@
         }
     }],
 
+    "sampling_rate": 100,
+    "frame_size": 800,
+
     "num_classes": 4,
     "class_map": {
         "0": 0,
@@ -24,26 +27,22 @@
         "SR", "SB", "AFIB", "GSVT"
     ],
 
-
-    "sampling_rate": 100,
-    "frame_size": 800,
-
     "samples_per_patient": [5, 5, 5, 10],
     "val_samples_per_patient": [5, 5, 5, 10],
     "test_samples_per_patient": [5, 5, 5, 10],
 
     "val_size": 40000,
-    "test_size": 50000,
+    "test_size": 40000,
     "val_file": "val.tfds",
     "test_file": "val.tfds",
 
-    "val_patients": 0.20,
     "batch_size": 256,
     "buffer_size": 50000,
-    "epochs": 150,
+    "epochs": 200,
     "steps_per_epoch": 50,
     "class_weights": "balanced",
     "val_metric": "loss",
+    "val_patients": 0.20,
     "lr_rate": 1e-3,
     "lr_cycles": 1,
 
@@ -121,7 +120,7 @@
     }],
 
     "model_file": "model.keras",
-    "use_logits": true,
+    "use_logits": false,
     "architecture": {
         "name": "efficientnetv2",
         "params": {