End of training

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: distilbert-base-uncased
+base_model: distilbert-base-uncased-finetuned-sst-2-english
 tags:
 - generated_from_trainer
 metrics:
@@ -16,11 +16,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # test_trainer
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
+This model is a fine-tuned version of [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4547
-- Accuracy: 0.9067
-- F1: 0.9389
+- Loss: 0.5228
+- Accuracy: 0.912
+- F1: 0.9426
 
 ## Model description
 

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "distilbert-base-uncased",
+  "_name_or_path": "distilbert-base-uncased-finetuned-sst-2-english",
   "activation": "gelu",
   "architectures": [
     "DistilBertForSequenceClassification"
@@ -7,12 +7,22 @@
   "attention_dropout": 0.1,
   "dim": 768,
   "dropout": 0.1,
+  "finetuning_task": "sst-2",
   "hidden_dim": 3072,
+  "id2label": {
+    "0": "NEGATIVE",
+    "1": "POSITIVE"
+  },
   "initializer_range": 0.02,
+  "label2id": {
+    "NEGATIVE": 0,
+    "POSITIVE": 1
+  },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
   "n_layers": 6,
+  "output_past": true,
   "pad_token_id": 0,
   "problem_type": "single_label_classification",
   "qa_dropout": 0.1,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:134aa45a87808aac2cef9026ce453f7a4ed92777051924c34b0936bedeffa8b9
+oid sha256:50d6d02d800ab640ba5d4022ea411cdbc9d1a8f43c868611d36ed66c8a06b491
 size 267832560

runs/Jun16_20-49-39_80beb8947a73/events.out.tfevents.1718570979.80beb8947a73.4231.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e71f0887a35a9811046330ddfdae6aea8c427d5fe9b8be3d8d4cb862e87be518
+size 5870

runs/Jun16_20-49-39_80beb8947a73/events.out.tfevents.1718571373.80beb8947a73.4231.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dad4dc380121e908e520430b435886ce4e13d09fd0749b7cb10c6e99b3a7c902
+size 457

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8dd1043588344bdeb19846e938c460c632a10bc98c41f7ae666da0c04b7a9af0
+oid sha256:f46468a2c330dc2bdf0844a6d121132bb352e75877e62826c86c84ea5feaebfd
 size 5112