End of training

README.md

@@ -1,24 +1,24 @@
 ---
 license: apache-2.0
-base_model: sentence-transformers/all-mpnet-base-v2
+base_model: distilbert/distilbert-base-uncased
 tags:
 - generated_from_trainer
 metrics:
 - accuracy
 model-index:
-- name: MPNetForClassification
+- name: DistilBERTForClassification
   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. -->
 
-# MPNetForClassification
+# DistilBERTForClassification
 
-This model is a fine-tuned version of [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) on the None dataset.
+This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3608
-- Accuracy: 0.84
+- Loss: 0.3499
+- Accuracy: 0.857
 
 ## Model description
 
@@ -49,8 +49,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.5352        | 1.0   | 625  | 0.3941          | 0.835    |
-| 0.4063        | 2.0   | 1250 | 0.3608          | 0.84     |
+| 0.5119        | 1.0   | 625  | 0.3677          | 0.832    |
+| 0.3729        | 2.0   | 1250 | 0.3499          | 0.857    |
 
 
 ### Framework versions