Checkpoint at epoch 2

epoch_2/metadata.json

@@ -1,48 +1,21 @@
 {
   "epoch": 2,
-  "loss": 0.034092439424611425,
-  "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
-    },
-    {
-      "Epoch": 2,
-      "Train Loss": "0.0191",
-      "Val Loss": "0.0341",
-      "Accuracy": "90.73%",
-      "Balanced Acc": "94.45%",
-      "F1 Pos": "94.56%",
-      "F1 Neg": "68.68%",
-      "Recall Neg": "99.14%",
-      "TP": 20014,
-      "TN": 2526,
-      "FP": 22,
-      "FN": 2282
-    }
-  ]
+  "loss": 0.04033649780267929,
+  "results": {
+    "loss": 0.04033649780267929,
+    "accuracy": 0.8819835775237482,
+    "balanced_accuracy": 0.9318150005717283,
+    "auroc": 0.992337310224247,
+    "auprc": 0.9985192216838991,
+    "TP": 19378,
+    "TN": 2534,
+    "FP": 14,
+    "FN": 2918,
+    "precision_pos": 0.9992780528052805,
+    "recall_pos": 0.8691245066379619,
+    "f1_pos": 0.9296680099788907,
+    "precision_neg": 0.46478356566397655,
+    "recall_neg": 0.9945054945054945,
+    "f1_neg": 0.6335000000000001
+  }
 }

epoch_2/model_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cc5ea5ef513c58d39126b95b9e1aa95c0eb1116349878d51743abf79235b9047
-size 377987684
+oid sha256:38de10cf76db7ff5b49a2e8a2da9604fd04f4e16ede5eccd0394541dae9928fd
+size 377987876

epoch_2/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:202abc942e53a1f97928be302a7601577717ffedc2f14c2242393aecc72d88f8
-size 825449
+oid sha256:50ab132d9b737ebe3fe77c30f36445b36440b20643f3e3166535ad691665b045
+size 827237