Checkpoint at epoch 4

epoch_4/metadata.json

@@ -1,76 +1,21 @@
 {
   "epoch": 4,
-  "loss": 0.03385985795678164,
-  "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
-    }
-  ]
+  "loss": 0.04616409618415878,
+  "results": {
+    "loss": 0.04616409618415878,
+    "accuracy": 0.8722025438737724,
+    "balanced_accuracy": 0.9270608197737984,
+    "auroc": 0.9945869587381199,
+    "auprc": 0.9989395603982902,
+    "TP": 19131,
+    "TN": 2538,
+    "FP": 10,
+    "FN": 3165,
+    "precision_pos": 0.9994775612559428,
+    "recall_pos": 0.8580462863293864,
+    "f1_pos": 0.9233776576489611,
+    "precision_neg": 0.44502893214097844,
+    "recall_neg": 0.9960753532182104,
+    "f1_neg": 0.6151981577990546
+  }
 }

epoch_4/model_state_dict.pth

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

epoch_4/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f20024b14f746f8f1ffdd0569663f2c2775a3afc94bfd115b417560af0dbe882
-size 825449
+oid sha256:4ae6a5a81f5f8972ed850c7bf23189465ac703dbc930a7a434878ada410bf109
+size 827237