Training complete

README.md

@@ -18,13 +18,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # biogpt-disease-ner
 
-This model is a fine-tuned version of [microsoft/biogpt](https://huggingface.co/microsoft/biogpt) on an unknown dataset.
+This model is a fine-tuned version of [microsoft/biogpt](https://huggingface.co/microsoft/biogpt) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1467
-- Precision: 0.4374
-- Recall: 0.5495
-- F1: 0.4871
-- Accuracy: 0.9569
+- Loss: 0.1525
+- Precision: 0.5241
+- Recall: 0.6075
+- F1: 0.5628
+- Accuracy: 0.9522
 
 ## Model description
 
@@ -55,9 +55,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| 0.3216        | 1.0   | 680  | 0.1684          | 0.3173    | 0.4099 | 0.3577 | 0.9458   |
-| 0.1623        | 2.0   | 1360 | 0.1529          | 0.4064    | 0.4962 | 0.4469 | 0.9535   |
-| 0.0967        | 3.0   | 2040 | 0.1467          | 0.4374    | 0.5495 | 0.4871 | 0.9569   |
+| 0.27          | 1.0   | 593  | 0.1769          | 0.4632    | 0.4523 | 0.4577 | 0.9395   |
+| 0.1581        | 2.0   | 1186 | 0.1561          | 0.5337    | 0.5083 | 0.5207 | 0.9475   |
+| 0.1225        | 3.0   | 1779 | 0.1525          | 0.5241    | 0.6075 | 0.5628 | 0.9522   |
 
 
 ### Framework versions