End of training

README.md

@@ -3,26 +3,11 @@ license: apache-2.0
 base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
-datasets:
-- glue
 metrics:
 - accuracy
 model-index:
 - name: distilbert-base-uncased-finetuned-sst2
-  results:
-  - task:
-      name: Text Classification
-      type: text-classification
-    dataset:
-      name: glue
-      type: glue
-      config: sst2
-      split: validation
-      args: sst2
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.9071100917431193
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilbert-base-uncased-finetuned-sst2
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the glue dataset.
+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: 0.2605
-- Accuracy: 0.9071
+- Loss: 0.2763
+- Accuracy: 0.9025
 
 ## Model description
 
@@ -64,16 +49,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 66   | 0.2770          | 0.8796   |
-| No log        | 2.0   | 132  | 0.2601          | 0.8979   |
-| No log        | 3.0   | 198  | 0.2605          | 0.9071   |
-| No log        | 4.0   | 264  | 0.2786          | 0.9025   |
-| No log        | 5.0   | 330  | 0.2781          | 0.9048   |
+| No log        | 1.0   | 66   | 0.2804          | 0.8796   |
+| No log        | 2.0   | 132  | 0.2631          | 0.8968   |
+| No log        | 3.0   | 198  | 0.2641          | 0.8979   |
+| No log        | 4.0   | 264  | 0.2763          | 0.9025   |
+| No log        | 5.0   | 330  | 0.2811          | 0.9025   |
 
 
 ### Framework versions
 
-- Transformers 4.35.2
-- Pytorch 2.1.1+cu121
-- Datasets 2.15.0
-- Tokenizers 0.15.0
+- Transformers 4.39.1
+- Pytorch 2.2.1+cu121
+- Datasets 2.18.0
+- Tokenizers 0.15.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1159ea59e0b2626ca05b4c29400bdd16699d5ef823e20be15da9401c8c3a08a9
+oid sha256:0b617d57c9538f741b48c2d797c1c8bb1dd75888e51d9638ad40ef2b3b1f22bb
 size 267832560