End of training

README.md

@@ -3,6 +3,8 @@ license: apache-2.0
 base_model: distilbert-base-cased
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
 - name: distilbert_project_test_trainer
   results: []
@@ -14,6 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
 # distilbert_project_test_trainer
 
 This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0790
+- Accuracy: 0.9757
 
 ## Model description
 
@@ -40,9 +45,18 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 3.0
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| No log        | 1.0   | 227  | 0.0794          | 0.9779   |
+| No log        | 2.0   | 454  | 0.0666          | 0.9801   |
+| 0.2006        | 3.0   | 681  | 0.0790          | 0.9757   |
+
+
 ### Framework versions
 
 - Transformers 4.35.2
 - Pytorch 2.1.0+cu121
 - Datasets 2.16.1
-- Tokenizers 0.15.0
+- Tokenizers 0.15.1

config.json

@@ -25,6 +25,7 @@
   "n_layers": 6,
   "output_past": true,
   "pad_token_id": 0,
+  "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:3a386e42fea8f24c3c8497a855048028b41631639f9dbb95f399370327c29c68
+oid sha256:debc5fe67d8711206dc0c3960565f09724c086f72ee744d883844e7399b24544
 size 263147764

runs/Jan24_19-21-24_984caef51383/events.out.tfevents.1706124085.984caef51383.229.3

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

runs/Jan24_19-21-24_984caef51383/events.out.tfevents.1706124600.984caef51383.229.7

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e5ae8d2c27eac5529535b66c43790cb5773aee73437e020044ad13e30217af9b
+oid sha256:d525c94875356182420153eecf136422ef5541f17b15c96840698c690986e322
 size 4600