Training complete

README.md

@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [medicalai/ClinicalBERT](https://huggingface.co/medicalai/ClinicalBERT) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0654
-- Precision: 0.8341
-- Recall: 0.8261
-- F1: 0.8301
-- Accuracy: 0.9806
+- Loss: 0.0815
+- Precision: 0.7883
+- Recall: 0.7691
+- F1: 0.7786
+- Accuracy: 0.9751
 
 ## Model description
 
@@ -57,13 +57,13 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| No log        | 1.0   | 231  | 0.0808          | 0.7834    | 0.8124 | 0.7977 | 0.9763   |
-| No log        | 2.0   | 462  | 0.0654          | 0.8341    | 0.8261 | 0.8301 | 0.9806   |
+| No log        | 1.0   | 231  | 0.0931          | 0.7518    | 0.7606 | 0.7562 | 0.9719   |
+| No log        | 2.0   | 462  | 0.0815          | 0.7883    | 0.7691 | 0.7786 | 0.9751   |
 
 
 ### Framework versions
 
 - Transformers 4.44.2
-- Pytorch 2.4.1+cu121
+- Pytorch 2.4.0+cu121
 - Datasets 2.21.0
 - Tokenizers 0.19.1