davidadamczyk/all-mpnet-base-v2-sst5

README.md

@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [sentence-transformers/all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2158
+- Loss: 1.1440
 
 ## Model description
 
@@ -47,9 +47,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 0.3438        | 1.0   | 94   | 0.2678          |
-| 0.1819        | 2.0   | 188  | 0.2158          |
-| 0.1632        | 3.0   | 282  | 0.2162          |
+| 1.308         | 1.0   | 94   | 1.2669          |
+| 1.183         | 2.0   | 188  | 1.1781          |
+| 1.0672        | 3.0   | 282  | 1.1440          |
 
 
 ### Framework versions

config.json

@@ -9,8 +9,22 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
+  "id2label": {
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2",
+    "3": "LABEL_3",
+    "4": "LABEL_4"
+  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
+  "label2id": {
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2,
+    "LABEL_3": 3,
+    "LABEL_4": 4
+  },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,
   "model_type": "mpnet",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ece2b045b661d318d7a69a539cad084c26a2ea3dd3cd0edacd9ee0da1cdf0bcd
-size 437975200
+oid sha256:07b493becd2dd47cdebcde6cab92225f409ba502ffefffa816ccb8cb03daec1f
+size 437984428