Checkpoint at epoch 0

epoch_0/metadata.json

@@ -1,21 +1,21 @@
 {
   "epoch": 0,
-  "loss": 0.13854153530719954,
+  "loss": 0.11473902639006402,
   "results": {
-    "loss": 0.13854153530719954,
-    "accuracy": 0.6525764895330113,
-    "balanced_accuracy": 0.6692344232443987,
-    "auroc": 0.7319812806235101,
-    "auprc": 0.9534032459559243,
-    "TP": 1445,
-    "TN": 176,
-    "FP": 79,
-    "FN": 784,
-    "precision_pos": 0.9481627296587927,
-    "recall_pos": 0.6482727680574248,
-    "f1_pos": 0.770050626165734,
-    "precision_neg": 0.18333333333333332,
-    "recall_neg": 0.6901960784313725,
-    "f1_neg": 0.2897119341563786
+    "loss": 0.11473902639006402,
+    "accuracy": 0.7199726292062469,
+    "balanced_accuracy": 0.7643819575524173,
+    "auroc": 0.8294207002375347,
+    "auprc": 0.9742834369584057,
+    "TP": 15797,
+    "TN": 2090,
+    "FP": 458,
+    "FN": 6499,
+    "precision_pos": 0.9718240541371885,
+    "recall_pos": 0.7085127377108001,
+    "f1_pos": 0.819537755181448,
+    "precision_neg": 0.24333449761322623,
+    "recall_neg": 0.8202511773940345,
+    "f1_neg": 0.37532549160456136
   }
 }

epoch_0/model_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0506b13c1182af9ccef54eab3d972685ae9d2ca163b9a6521b736d604e34dd9c
+oid sha256:f993360f6b2112f8eb9346ca74902edb195ac45d75a2e18bfa35e1154d341a06
 size 1430561763

epoch_0/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1fc7a3439d604ca4832b5768fb111310376dd8ceac01ad0da5065eff59f36bd3
+oid sha256:a672d68336eb60ce38cd737f2621245d9b3811f568a3d70717b778bf0f81113b
 size 1055529