Checkpoint at epoch 0

epoch_0/metadata.json

@@ -1,21 +1,21 @@
 {
   "epoch": 0,
-  "loss": 0.09374230787484035,
+  "loss": 0.054390631873817444,
   "results": {
-    "loss": 0.09374230787484035,
-    "accuracy": 0.833883432619546,
-    "balanced_accuracy": 0.7866540463995485,
-    "auroc": 0.8786745684860016,
-    "auprc": 0.9829854592230153,
-    "TP": 18864,
-    "TN": 1853,
-    "FP": 695,
-    "FN": 3432,
-    "precision_pos": 0.9644664860166675,
-    "recall_pos": 0.8460710441334769,
-    "f1_pos": 0.9013976824752121,
-    "precision_neg": 0.35061494796594134,
-    "recall_neg": 0.7272370486656201,
-    "f1_neg": 0.47312651602195843
+    "loss": 0.054390631873817444,
+    "accuracy": 0.8993720817903719,
+    "balanced_accuracy": 0.8924893216374071,
+    "auroc": 0.959779191443904,
+    "auprc": 0.995259551598612,
+    "TP": 20092,
+    "TN": 2252,
+    "FP": 296,
+    "FN": 2204,
+    "precision_pos": 0.9854816558760054,
+    "recall_pos": 0.9011481880157876,
+    "f1_pos": 0.9414300440446068,
+    "precision_neg": 0.5053859964093357,
+    "recall_neg": 0.8838304552590267,
+    "f1_neg": 0.6430611079383208
   }
 }

epoch_0/model_state_dict.pth

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

epoch_0/optimizer_state_dict.pth

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