Checkpoint at epoch 17

epoch_17/metadata.json

@@ -1,34 +1,21 @@
 {
   "epoch": 17,
-  "loss": 0.05047423076910505,
-  "results": [
-    {
-      "Epoch": 16,
-      "Train Loss": "0.0101",
-      "Val Loss": "0.0302",
-      "Accuracy": "92.31%",
-      "Balanced Acc": "95.47%",
-      "F1 Pos": "95.53%",
-      "F1 Neg": "72.62%",
-      "Recall Neg": "99.45%",
-      "TP": 20399,
-      "TN": 2534,
-      "FP": 14,
-      "FN": 1897
-    },
-    {
-      "Epoch": 17,
-      "Train Loss": "0.0112",
-      "Val Loss": "0.0505",
-      "Accuracy": "87.68%",
-      "Balanced Acc": "92.94%",
-      "F1 Pos": "92.63%",
-      "F1 Neg": "62.36%",
-      "Recall Neg": "99.57%",
-      "TP": 19245,
-      "TN": 2537,
-      "FP": 11,
-      "FN": 3051
-    }
-  ]
+  "loss": 0.02203066671609122,
+  "results": {
+    "loss": 0.02203066671609122,
+    "accuracy": 0.9401062630816294,
+    "balanced_accuracy": 0.9640236839125813,
+    "auroc": 0.9975138534961887,
+    "auprc": 0.9996803449451499,
+    "TP": 20823,
+    "TN": 2533,
+    "FP": 15,
+    "FN": 1473,
+    "precision_pos": 0.9992801612438814,
+    "recall_pos": 0.9339343379978472,
+    "f1_pos": 0.965502851578801,
+    "precision_neg": 0.6323015476784822,
+    "recall_neg": 0.9941130298273155,
+    "f1_neg": 0.7729630759841317
+  }
 }

epoch_17/model_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b3cb3aa75069c8a276507cfd1a362180da9af3ea853fcdd4a30d0cd0fe0a7fe1
-size 377987684
+oid sha256:b4c7a0dbdad14007fc931907934f43b8cbd9b87733a1ffd8f61e5da607a39a7d
+size 377987876

epoch_17/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:90182837b18895b723a652db58e40578546c021eb048ff11c49988473a7e7dc4
-size 825449
+oid sha256:9b0b8334df018c2155e8b7dcaa374a994ab214d034929f694cd6fb7cc5e952f4
+size 827237