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.0918
+- Loss: 1.0984
 
 ## Model description
 
@@ -47,12 +47,12 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 1.0941        | 2.0   | 10   | 1.0918          |
+| 1.1442        | 2.0   | 10   | 1.0984          |
 
 
 ### Framework versions
 
-- Transformers 4.41.2
-- Pytorch 2.3.0+cu121
+- Transformers 4.42.4
+- Pytorch 2.3.1+cu121
 - Datasets 2.20.0
 - Tokenizers 0.19.1

config.json

@@ -9,27 +9,23 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2"
+    "0": "LABEL_0"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2
+    "LABEL_0": 0
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,
-  "problem_type": "single_label_classification",
+  "problem_type": "regression",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.41.2",
+  "transformers_version": "4.42.4",
   "vocab_size": 30522
 }

model.safetensors

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

training_args.bin

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