Checkpoint at epoch 1

epoch_1/metadata.json

@@ -1,34 +1,21 @@
 {
   "epoch": 1,
-  "loss": 0.048302005067314684,
-  "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
-    }
-  ]
+  "loss": 0.03908254242558488,
+  "results": {
+    "loss": 0.03908254242558488,
+    "accuracy": 0.8812993076799227,
+    "balanced_accuracy": 0.931086152685799,
+    "auroc": 0.990585626794396,
+    "auprc": 0.99832702823937,
+    "TP": 19363,
+    "TN": 2532,
+    "FP": 16,
+    "FN": 2933,
+    "precision_pos": 0.9991743640022706,
+    "recall_pos": 0.8684517402224614,
+    "f1_pos": 0.929238152369526,
+    "precision_neg": 0.4633119853613907,
+    "recall_neg": 0.9937205651491365,
+    "f1_neg": 0.6319730438038188
+  }
 }

epoch_1/model_state_dict.pth

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

epoch_1/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a4517bbe32315f79f90187702e748b35529f626c715386ff1b012d632b3d2b5a
-size 825449
+oid sha256:85c1dfe6d009da202aba1a448760f54db1334abfe456c9b6255cb0fd1d411177
+size 827237