Checkpoint at epoch 20

epoch_20/metadata.json

@@ -1,76 +1,21 @@
 {
   "epoch": 20,
-  "loss": 0.05384415975917423,
-  "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
-    }
-  ]
+  "loss": 0.018804812356772335,
+  "results": {
+    "loss": 0.018804812356772335,
+    "accuracy": 0.9500885525680245,
+    "balanced_accuracy": 0.9692376060302402,
+    "auroc": 0.9977244934572321,
+    "auprc": 0.9997046711356998,
+    "TP": 21073,
+    "TN": 2531,
+    "FP": 17,
+    "FN": 1223,
+    "precision_pos": 0.9991939307728781,
+    "recall_pos": 0.9451471115895228,
+    "f1_pos": 0.9714193518646569,
+    "precision_neg": 0.6742141715503462,
+    "recall_neg": 0.9933281004709577,
+    "f1_neg": 0.8032370675975881
+  }
 }

epoch_20/model_state_dict.pth

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

epoch_20/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:002dacf281a7da3eb2cbd233fecc6e082349c33bf81755ae55992bb797a51fda
-size 825449
+oid sha256:bf9e5d7e5042ab27879e4444a9a6869e2b7e3e4ad9229d17a0f4b55034a6666e
+size 827237