Checkpoint at epoch 5

epoch_5/metadata.json

@@ -1,21 +1,21 @@
 {
   "epoch": 5,
-  "loss": 0.04623098404145532,
+  "loss": 0.02193514314356369,
   "results": {
-    "loss": 0.04623098404145532,
-    "accuracy": 0.8884640154564483,
-    "balanced_accuracy": 0.9102235288418588,
-    "auroc": 0.966964634243198,
-    "auprc": 0.995591372196887,
-    "TP": 19684,
-    "TN": 2389,
-    "FP": 159,
-    "FN": 2612,
-    "precision_pos": 0.9919870987249911,
-    "recall_pos": 0.8828489415141729,
-    "f1_pos": 0.9342414390469636,
-    "precision_neg": 0.4777044591081784,
-    "recall_neg": 0.9375981161695447,
-    "f1_neg": 0.6329315141078289
+    "loss": 0.02193514314356369,
+    "accuracy": 0.9700128803735308,
+    "balanced_accuracy": 0.9624361523196676,
+    "auroc": 0.9934773870217126,
+    "auprc": 0.999072924960508,
+    "TP": 21671,
+    "TN": 2428,
+    "FP": 120,
+    "FN": 625,
+    "precision_pos": 0.9944931393694645,
+    "recall_pos": 0.9719680660208109,
+    "f1_pos": 0.9831015945743644,
+    "precision_neg": 0.7952833278742221,
+    "recall_neg": 0.9529042386185244,
+    "f1_neg": 0.8669880378503838
   }
 }

epoch_5/model_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:080ae5556884134f4588c0c66eacadbc25a0237aabe751d8c104e12e0929ea5a
+oid sha256:d8a7d856db420e9b26dd9455fcb218085cab7a828ad6dd301982f88e3fded1c9
 size 377988324

epoch_5/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f7deeea3af34fe246c6a08bc3848d73e3ca2b9a06151c3423030b48f9ab232a5
+oid sha256:c87f34786db361c9e0b83f52147f3640d173970b96c96ca0c2019e10b2fad506
 size 827237