Checkpoint at epoch 21

epoch_21/metadata.json

@@ -1,90 +1,21 @@
 {
   "epoch": 21,
-  "loss": 0.028453392107795163,
-  "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
-    },
-    {
-      "Epoch": 18,
-      "Train Loss": "0.0103",
-      "Val Loss": "0.0196",
-      "Accuracy": "95.73%",
-      "Balanced Acc": "97.13%",
-      "F1 Pos": "97.56%",
-      "F1 Neg": "82.60%",
-      "Recall Neg": "98.90%",
-      "TP": 21262,
-      "TN": 2520,
-      "FP": 28,
-      "FN": 1034
-    },
-    {
-      "Epoch": 19,
-      "Train Loss": "0.0093",
-      "Val Loss": "0.0257",
-      "Accuracy": "93.52%",
-      "Balanced Acc": "96.08%",
-      "F1 Pos": "96.26%",
-      "F1 Neg": "75.86%",
-      "Recall Neg": "99.29%",
-      "TP": 20704,
-      "TN": 2530,
-      "FP": 18,
-      "FN": 1592
-    },
-    {
-      "Epoch": 20,
-      "Train Loss": "0.0097",
-      "Val Loss": "0.0538",
-      "Accuracy": "87.26%",
-      "Balanced Acc": "92.74%",
-      "F1 Pos": "92.36%",
-      "F1 Neg": "61.60%",
-      "Recall Neg": "99.65%",
-      "TP": 19139,
-      "TN": 2539,
-      "FP": 9,
-      "FN": 3157
-    },
-    {
-      "Epoch": 21,
-      "Train Loss": "0.0107",
-      "Val Loss": "0.0285",
-      "Accuracy": "92.61%",
-      "Balanced Acc": "95.60%",
-      "F1 Pos": "95.71%",
-      "F1 Neg": "73.39%",
-      "Recall Neg": "99.37%",
-      "TP": 20476,
-      "TN": 2532,
-      "FP": 16,
-      "FN": 1820
-    }
-  ]
+  "loss": 0.01938563861829799,
+  "results": {
+    "loss": 0.01938563861829799,
+    "accuracy": 0.9478747383674127,
+    "balanced_accuracy": 0.9685256213108743,
+    "auroc": 0.99786030531696,
+    "auprc": 0.9997214104850493,
+    "TP": 21015,
+    "TN": 2534,
+    "FP": 14,
+    "FN": 1281,
+    "precision_pos": 0.9993342526986543,
+    "recall_pos": 0.942545748116254,
+    "f1_pos": 0.9701096364685515,
+    "precision_neg": 0.6642201834862386,
+    "recall_neg": 0.9945054945054945,
+    "f1_neg": 0.7964796479647966
+  }
 }

epoch_21/model_state_dict.pth

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

epoch_21/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d6cfb1a1310c62c9237a1b1b774033f96f6597a6acd9d0bba608ed5ffbd775d3
-size 825449
+oid sha256:54735fb08793abe624e290f16fa0a049f819ace6bb9a895b6f3604735b3825ab
+size 827237