End of training

README.md

@@ -3,6 +3,8 @@ license: apache-2.0
 base_model: albert-base-v2
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
 - name: albert_project_test_trainer
   results: []
@@ -14,6 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
 # albert_project_test_trainer
 
 This model is a fine-tuned version of [albert-base-v2](https://huggingface.co/albert-base-v2) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.2782
+- Accuracy: 0.9447
 
 ## 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.4055          | 0.8584   |
+| No log        | 2.0   | 454  | 0.2235          | 0.9336   |
+| 0.4752        | 3.0   | 681  | 0.2782          | 0.9447   |
+
+
 ### 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

@@ -36,6 +36,7 @@
   "num_memory_blocks": 0,
   "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:bb460050c64e3cee1152ea58072091636f5bf32b36c043c33d4fc7b255695629
+oid sha256:2f3ab1488aa418760cea39bab8cc42718725f41cbbb83e490227b8f21d99c076
 size 46746988

runs/Jan24_19-06-24_984caef51383/events.out.tfevents.1706123184.984caef51383.229.2

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

runs/Jan24_19-06-24_984caef51383/events.out.tfevents.1706124593.984caef51383.229.6

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

training_args.bin

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