MadhuriD/Product_select

README.md

@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.0491
+- Loss: 1.0918
 
 ## Model description
 
@@ -47,7 +47,7 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 1.0593        | 2.0   | 10   | 1.0491          |
+| 1.0941        | 2.0   | 10   | 1.0918          |
 
 
 ### Framework versions

config.json

@@ -9,18 +9,22 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "LABEL_0"
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "LABEL_0": 0
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,
-  "problem_type": "regression",
+  "problem_type": "single_label_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:edfe01676cdb6679e40ba6a0fd5649c0741b757f2a59c393f773a4678e1155ea
-size 267829484
+oid sha256:f5d9ddd71f1ff16670121c8defebc5d85061e3a8fe9581b15a89c29d0e9cc66a
+size 267835644

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f3438f997dc2bdde171a317c1cbf4587c3802633270be7d34570ae11bf11132d
+oid sha256:def96b3802b6167f441a9432f84321105ce1af39d40729bd5429d7bc342b013b
 size 5048