Checkpoint at epoch 4

epoch_4/metadata.json

@@ -1,21 +1,21 @@
 {
   "epoch": 4,
-  "loss": 0.0518628698318209,
+  "loss": 0.02321032434574567,
   "results": {
-    "loss": 0.0518628698318209,
-    "accuracy": 0.9040814683625825,
-    "balanced_accuracy": 0.8998058940393248,
-    "auroc": 0.9617851865636542,
-    "auprc": 0.9948404547018816,
-    "TP": 20182,
-    "TN": 2279,
-    "FP": 269,
-    "FN": 2114,
-    "precision_pos": 0.9868466089677767,
-    "recall_pos": 0.9051847865087909,
-    "f1_pos": 0.9442533978992679,
-    "precision_neg": 0.5187798770771682,
-    "recall_neg": 0.8944270015698587,
-    "f1_neg": 0.656677712145224
+    "loss": 0.02321032434574567,
+    "accuracy": 0.9636531959426823,
+    "balanced_accuracy": 0.960109563408041,
+    "auroc": 0.9923030998231867,
+    "auprc": 0.9989196200362221,
+    "TP": 21506,
+    "TN": 2435,
+    "FP": 113,
+    "FN": 790,
+    "precision_pos": 0.9947731162403441,
+    "recall_pos": 0.964567635450305,
+    "f1_pos": 0.979437549812137,
+    "precision_neg": 0.7550387596899225,
+    "recall_neg": 0.9556514913657771,
+    "f1_neg": 0.8435821929672614
   }
 }

epoch_4/model_state_dict.pth

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

epoch_4/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:693385f147bbf0092a039b661f23ee4cbaef2b14c1ba03e87e4f32f9c47ba122
+oid sha256:749e5f41a499ba6e94c3d3df17bdd2428359c5f635e84e0c6933f46b4a5d09c1
 size 827237