End of training

README.md

@@ -3,6 +3,8 @@ license: apache-2.0
 base_model: bert-base-cased
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
 - name: bert_project_test_trainer
   results: []
@@ -14,6 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
 # bert_project_test_trainer
 
 This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0511
+- Accuracy: 0.9845
 
 ## 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.1338          | 0.9624   |
+| No log        | 2.0   | 454  | 0.0853          | 0.9801   |
+| 0.1808        | 3.0   | 681  | 0.0511          | 0.9845   |
+
+
 ### 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

@@ -28,6 +28,7 @@
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
+  "problem_type": "single_label_classification",
   "torch_dtype": "float32",
   "transformers_version": "4.35.2",
   "type_vocab_size": 2,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:23a5029d0de578589c783c4c28a14954127cee86832515b8094a7773b98a28d5
+oid sha256:520ec11f0904a326697283799d0385654c48af79367800ada92d13d75338719b
 size 433273844

runs/Jan24_18-44-59_984caef51383/events.out.tfevents.1706121900.984caef51383.229.0

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

runs/Jan24_18-44-59_984caef51383/events.out.tfevents.1706124562.984caef51383.229.4

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

training_args.bin

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