Update README.md

README.md

@@ -23,3 +23,33 @@ Note, we are only training 0.58% of the parameters, using only the query, key, a
 ```
 trainable params: 23682 || all params: 4075265 || trainable%: 0.5811155838945443
 ```
+
+## Testing for Overfitting
+
+### Checkpoint 1
+
+### Checkpoint 2
+
+### Checkpoint 3
+
+### Checkpoint 4
+
+```python
+Train metrics:
+{'eval_loss': 0.24070295691490173,
+'eval_accuracy': 0.9018779246397052,
+'eval_precision': 0.16624103834249204,
+'eval_recall': 0.8651772818812425,
+'eval_f1': 0.27889357183237473,
+'eval_auc': 0.8839390799308487,
+'eval_mcc': 0.3536803490333407}
+
+Test metrics:
+{'eval_loss': 0.26776671409606934,
+'eval_accuracy': 0.8902711124906878,
+'eval_precision': 0.13008662855482372,
+'eval_recall': 0.7084623832213568,
+'eval_f1': 0.219811797752809,
+'eval_auc': 0.8013943890942485,
+'eval_mcc': 0.2721459410994918}
+```