End of training

README.md

@@ -7,19 +7,19 @@ tags:
 metrics:
 - f1
 model-index:
-- name: minilm-finetuned-emotion
+- name: minilm-language-classifier
   results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# minilm-finetuned-emotion
+# minilm-language-classifier
 
 This model is a fine-tuned version of [microsoft/MiniLM-L12-H384-uncased](https://huggingface.co/microsoft/MiniLM-L12-H384-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.9498
-- F1: 0.6330
+- Loss: 0.4820
+- F1: 0.8756
 
 ## Model description
 
@@ -51,8 +51,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:------:|
-| 1.4003        | 1.0   | 250  | 1.0991          | 0.5758 |
-| 1.0246        | 2.0   | 500  | 0.9498          | 0.6330 |
+| 0.7743        | 1.0   | 250  | 0.5844          | 0.8507 |
+| 0.5297        | 2.0   | 500  | 0.4820          | 0.8756 |
 
 
 ### Framework versions