Model save

README.md

@@ -8,6 +8,9 @@ datasets:
 - emotion
 metrics:
 - accuracy
+- f1
+- precision
+- recall
 model-index:
 - name: emotion-gpt2-lora
   results: []
@@ -20,8 +23,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [openai-community/gpt2](https://huggingface.co/openai-community/gpt2) on the emotion dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1445
-- Accuracy: 0.9315
+- Loss: 0.1521
+- Accuracy: 0.933
+- F1: 0.9334
+- Precision: 0.9347
+- Recall: 0.933
 
 ## Model description
 
@@ -50,12 +56,12 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 250  | 0.3155          | 0.8815   |
-| 0.6098        | 2.0   | 500  | 0.1879          | 0.926    |
-| 0.6098        | 3.0   | 750  | 0.1576          | 0.934    |
-| 0.1844        | 4.0   | 1000 | 0.1445          | 0.9315   |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
+| No log        | 1.0   | 250  | 0.3191          | 0.8895   | 0.8902 | 0.8933    | 0.8895 |
+| 0.6939        | 2.0   | 500  | 0.1939          | 0.935    | 0.9349 | 0.9352    | 0.935  |
+| 0.6939        | 3.0   | 750  | 0.1689          | 0.931    | 0.9315 | 0.9329    | 0.931  |
+| 0.1897        | 4.0   | 1000 | 0.1521          | 0.933    | 0.9334 | 0.9347    | 0.933  |
 
 
 ### Framework versions