Checkpoint at epoch 14

epoch_14/metadata.json

@@ -1,216 +1,21 @@
 {
   "epoch": 14,
-  "loss": 0.029108050524671435,
-  "results": [
-    {
-      "Epoch": 1,
-      "Train Loss": "0.0664",
-      "Val Loss": "0.0429",
-      "Accuracy": "87.86%",
-      "Balanced Acc": "92.47%",
-      "F1 Pos": "92.76%",
-      "F1 Neg": "62.41%",
-      "Recall Neg": "98.27%",
-      "TP": 19324,
-      "TN": 2504,
-      "FP": 44,
-      "FN": 2972
-    },
-    {
-      "Epoch": 1,
-      "Train Loss": "0.0263",
-      "Val Loss": "0.0483",
-      "Accuracy": "86.24%",
-      "Balanced Acc": "92.00%",
-      "F1 Pos": "91.70%",
-      "F1 Neg": "59.67%",
-      "Recall Neg": "99.25%",
-      "TP": 18896,
-      "TN": 2529,
-      "FP": 19,
-      "FN": 3400
-    },
-    {
-      "Epoch": 2,
-      "Train Loss": "0.0191",
-      "Val Loss": "0.0341",
-      "Accuracy": "90.73%",
-      "Balanced Acc": "94.45%",
-      "F1 Pos": "94.56%",
-      "F1 Neg": "68.68%",
-      "Recall Neg": "99.14%",
-      "TP": 20014,
-      "TN": 2526,
-      "FP": 22,
-      "FN": 2282
-    },
-    {
-      "Epoch": 3,
-      "Train Loss": "0.0162",
-      "Val Loss": "0.0397",
-      "Accuracy": "89.09%",
-      "Balanced Acc": "93.73%",
-      "F1 Pos": "93.53%",
-      "F1 Neg": "65.18%",
-      "Recall Neg": "99.57%",
-      "TP": 19597,
-      "TN": 2537,
-      "FP": 11,
-      "FN": 2699
-    },
-    {
-      "Epoch": 4,
-      "Train Loss": "0.0148",
-      "Val Loss": "0.0339",
-      "Accuracy": "91.06%",
-      "Balanced Acc": "94.67%",
-      "F1 Pos": "94.76%",
-      "F1 Neg": "69.48%",
-      "Recall Neg": "99.22%",
-      "TP": 20095,
-      "TN": 2528,
-      "FP": 20,
-      "FN": 2201
-    },
-    {
-      "Epoch": 5,
-      "Train Loss": "0.0136",
-      "Val Loss": "0.0331",
-      "Accuracy": "91.21%",
-      "Balanced Acc": "94.82%",
-      "F1 Pos": "94.85%",
-      "F1 Neg": "69.86%",
-      "Recall Neg": "99.37%",
-      "TP": 20127,
-      "TN": 2532,
-      "FP": 16,
-      "FN": 2169
-    },
-    {
-      "Epoch": 6,
-      "Train Loss": "0.0123",
-      "Val Loss": "0.0364",
-      "Accuracy": "90.75%",
-      "Balanced Acc": "94.44%",
-      "F1 Pos": "94.57%",
-      "F1 Neg": "68.72%",
-      "Recall Neg": "99.10%",
-      "TP": 20020,
-      "TN": 2525,
-      "FP": 23,
-      "FN": 2276
-    },
-    {
-      "Epoch": 7,
-      "Train Loss": "0.0142",
-      "Val Loss": "0.0332",
-      "Accuracy": "91.42%",
-      "Balanced Acc": "94.94%",
-      "F1 Pos": "94.98%",
-      "F1 Neg": "70.37%",
-      "Recall Neg": "99.37%",
-      "TP": 20180,
-      "TN": 2532,
-      "FP": 16,
-      "FN": 2116
-    },
-    {
-      "Epoch": 8,
-      "Train Loss": "0.0130",
-      "Val Loss": "0.0810",
-      "Accuracy": "80.43%",
-      "Balanced Acc": "88.94%",
-      "F1 Pos": "87.77%",
-      "F1 Neg": "51.09%",
-      "Recall Neg": "99.65%",
-      "TP": 17444,
-      "TN": 2539,
-      "FP": 9,
-      "FN": 4852
-    },
-    {
-      "Epoch": 9,
-      "Train Loss": "0.0139",
-      "Val Loss": "0.0316",
-      "Accuracy": "91.68%",
-      "Balanced Acc": "95.09%",
-      "F1 Pos": "95.15%",
-      "F1 Neg": "71.02%",
-      "Recall Neg": "99.37%",
-      "TP": 20246,
-      "TN": 2532,
-      "FP": 16,
-      "FN": 2050
-    },
-    {
-      "Epoch": 10,
-      "Train Loss": "0.0134",
-      "Val Loss": "0.0274",
-      "Accuracy": "92.66%",
-      "Balanced Acc": "95.60%",
-      "F1 Pos": "95.74%",
-      "F1 Neg": "73.50%",
-      "Recall Neg": "99.29%",
-      "TP": 20490,
-      "TN": 2530,
-      "FP": 18,
-      "FN": 1806
-    },
-    {
-      "Epoch": 11,
-      "Train Loss": "0.0112",
-      "Val Loss": "0.0245",
-      "Accuracy": "93.85%",
-      "Balanced Acc": "96.19%",
-      "F1 Pos": "96.46%",
-      "F1 Neg": "76.78%",
-      "Recall Neg": "99.14%",
-      "TP": 20790,
-      "TN": 2526,
-      "FP": 22,
-      "FN": 1506
-    },
-    {
-      "Epoch": 12,
-      "Train Loss": "0.0120",
-      "Val Loss": "0.0255",
-      "Accuracy": "93.40%",
-      "Balanced Acc": "95.97%",
-      "F1 Pos": "96.19%",
-      "F1 Neg": "75.51%",
-      "Recall Neg": "99.22%",
-      "TP": 20676,
-      "TN": 2528,
-      "FP": 20,
-      "FN": 1620
-    },
-    {
-      "Epoch": 13,
-      "Train Loss": "0.0124",
-      "Val Loss": "0.0387",
-      "Accuracy": "89.88%",
-      "Balanced Acc": "94.15%",
-      "F1 Pos": "94.03%",
-      "F1 Neg": "66.86%",
-      "Recall Neg": "99.53%",
-      "TP": 19794,
-      "TN": 2536,
-      "FP": 12,
-      "FN": 2502
-    },
-    {
-      "Epoch": 14,
-      "Train Loss": "0.0107",
-      "Val Loss": "0.0291",
-      "Accuracy": "92.46%",
-      "Balanced Acc": "95.43%",
-      "F1 Pos": "95.62%",
-      "F1 Neg": "72.96%",
-      "Recall Neg": "99.18%",
-      "TP": 20444,
-      "TN": 2527,
-      "FP": 21,
-      "FN": 1852
-    }
-  ]
+  "loss": 0.0203186103307214,
+  "results": {
+    "loss": 0.0203186103307214,
+    "accuracy": 0.9449364031556915,
+    "balanced_accuracy": 0.9668885563664896,
+    "auroc": 0.9975325913258406,
+    "auprc": 0.9996790382013356,
+    "TP": 20942,
+    "TN": 2534,
+    "FP": 14,
+    "FN": 1354,
+    "precision_pos": 0.9993319335751097,
+    "recall_pos": 0.9392716182274847,
+    "f1_pos": 0.9683714047905299,
+    "precision_neg": 0.6517489711934157,
+    "recall_neg": 0.9945054945054945,
+    "f1_neg": 0.7874456183965196
+  }
 }

epoch_14/model_state_dict.pth

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

epoch_14/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a64c1e6b0ee0c40c14fb45c4c5b2930d4f54c7784d13bda479b4cee1bfa99637
-size 825449
+oid sha256:2d1f34cd63e54055e877189a223e9a98a9b257d0b1cbe95ba327884febcdc9c0
+size 827237