End of training

README.md

@@ -1,11 +1,7 @@
 ---
 library_name: transformers
-license: apache-2.0
-base_model: albert/albert-base-v2
 tags:
 - generated_from_trainer
-metrics:
-- accuracy
 model-index:
 - name: albert-category-classification-rob-dset
   results: []
@@ -16,10 +12,16 @@ should probably proofread and complete it, then remove this comment. -->
 
 # albert-category-classification-rob-dset
 
-This model is a fine-tuned version of [albert/albert-base-v2](https://huggingface.co/albert/albert-base-v2) on an unknown dataset.
+This model was trained from scratch on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.4822
-- Accuracy: 0.57
+- eval_loss: 1.1897
+- eval_model_preparation_time: 0.0007
+- eval_accuracy: 0.5857
+- eval_f1: 0.5732
+- eval_runtime: 94.6889
+- eval_samples_per_second: 31.683
+- eval_steps_per_second: 1.985
+- step: 0
 
 ## Model description
 
@@ -46,18 +48,6 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 10
 
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 438  | 1.3876          | 0.4887   |
-| 1.546         | 2.0   | 876  | 1.2214          | 0.5553   |
-| 1.0718        | 3.0   | 1314 | 1.1897          | 0.5857   |
-| 0.8159        | 4.0   | 1752 | 1.2674          | 0.5743   |
-| 0.5788        | 5.0   | 2190 | 1.3638          | 0.577    |
-| 0.3555        | 6.0   | 2628 | 1.4822          | 0.57     |
-
-
 ### Framework versions
 
 - Transformers 4.44.2

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "albert/albert-base-v2",
+  "_name_or_path": "pppereira3/albert-category-classification-rob-dset",
   "architectures": [
     "AlbertForSequenceClassification"
   ],

runs/Nov03_17-10-06_4c92b04c4d68/events.out.tfevents.1730653905.4c92b04c4d68.368.1

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

special_tokens_map.json

@@ -1,7 +1,25 @@
 {
-  "bos_token": "[CLS]",
-  "cls_token": "[CLS]",
-  "eos_token": "[SEP]",
+  "bos_token": {
+    "content": "[CLS]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "cls_token": {
+    "content": "[CLS]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "[SEP]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
   "mask_token": {
     "content": "[MASK]",
     "lstrip": true,
@@ -9,7 +27,25 @@
     "rstrip": false,
     "single_word": false
   },
-  "pad_token": "<pad>",
-  "sep_token": "[SEP]",
-  "unk_token": "<unk>"
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "[SEP]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
 }

tokenizer_config.json

@@ -48,10 +48,14 @@
   "eos_token": "[SEP]",
   "keep_accents": false,
   "mask_token": "[MASK]",
+  "max_length": 512,
   "model_max_length": 512,
   "pad_token": "<pad>",
   "remove_space": true,
   "sep_token": "[SEP]",
+  "stride": 0,
   "tokenizer_class": "AlbertTokenizer",
+  "truncation_side": "right",
+  "truncation_strategy": "longest_first",
   "unk_token": "<unk>"
 }

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0171aee8bc74b19edd3bec22a6434e9bf0c75006fb1c9da3683d0151fc2936d2
+oid sha256:55655abfe0960b256253c07497b76c7a4b637f08688809c209963f82f548d543
 size 5240