FrinzTheCoder/xlm-roberta-base-orm

README.md

@@ -6,7 +6,8 @@ tags:
 - generated_from_trainer
 metrics:
 - accuracy
-- f1
+- precision
+- recall
 model-index:
 - name: xlm-roberta-base-orm
   results: []
@@ -19,9 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [xlm-roberta-base](https://huggingface.co/xlm-roberta-base) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1154
-- Accuracy: 0.8671
-- F1: 0.8053
+- Loss: 0.1489
+- Accuracy: 0.7726
+- F1 Binary: 0.3856
+- Precision: 0.3070
+- Recall: 0.5185
 
 ## Model description
 
@@ -40,27 +43,28 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0003
-- train_batch_size: 16
+- learning_rate: 3e-05
+- train_batch_size: 32
 - eval_batch_size: 8
 - seed: 42
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
+- lr_scheduler_warmup_steps: 51
 - num_epochs: 4
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
-| 0.1428        | 1.0   | 861  | 0.1196          | 0.8671   | 0.8053 |
-| 0.122         | 2.0   | 1722 | 0.1154          | 0.8671   | 0.8053 |
-| 0.1215        | 3.0   | 2583 | 0.1153          | 0.8671   | 0.8053 |
-| 0.1173        | 4.0   | 3444 | 0.1154          | 0.8671   | 0.8053 |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Binary | Precision | Recall |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:---------:|:------:|
+| 0.1794        | 1.0   | 517  | 0.1732          | 0.3619   | 0.2385    | 0.1427    | 0.7258 |
+| 0.1723        | 2.0   | 1034 | 0.1690          | 0.5982   | 0.2995    | 0.1970    | 0.6239 |
+| 0.1522        | 3.0   | 1551 | 0.1818          | 0.8566   | 0.2847    | 0.4538    | 0.2074 |
+| 0.1436        | 4.0   | 2068 | 0.1489          | 0.7726   | 0.3856    | 0.3070    | 0.5185 |
 
 
 ### Framework versions
 
-- Transformers 4.44.2
-- Pytorch 2.4.1+cu121
-- Datasets 3.2.0
-- Tokenizers 0.19.1
+- Transformers 4.48.0
+- Pytorch 2.5.1+cu124
+- Datasets 3.1.0
+- Tokenizers 0.21.0

config.json

@@ -10,14 +10,8 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
-  "id2label": {
-    "0": "LABEL_0"
-  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
-  "label2id": {
-    "LABEL_0": 0
-  },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,
   "model_type": "xlm-roberta",
@@ -26,9 +20,8 @@
   "output_past": true,
   "pad_token_id": 1,
   "position_embedding_type": "absolute",
-  "problem_type": "regression",
   "torch_dtype": "float32",
-  "transformers_version": "4.44.2",
+  "transformers_version": "4.48.0",
   "type_vocab_size": 1,
   "use_cache": true,
   "vocab_size": 250002

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:17e6b5fb6411be87d769afa8a1e980cc8b1a3a5130f3639d0c3f21896f09e281
-size 1112201932
+oid sha256:1b64f71fff74d71c57fbcae41cab43d3af432f35f1a1d6421f875d67888ea627
+size 1112205008

tokenizer.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3ffb37461c391f096759f4a9bbbc329da0f36952f88bab061fcf84940c022e98
-size 17082999
+oid sha256:883b037111086fd4dfebbbc9b7cee11e1517b5e0c0514879478661440f137085
+size 17082987

tokenizer_config.json

@@ -42,9 +42,10 @@
     }
   },
   "bos_token": "<s>",
-  "clean_up_tokenization_spaces": true,
+  "clean_up_tokenization_spaces": false,
   "cls_token": "<s>",
   "eos_token": "</s>",
+  "extra_special_tokens": {},
   "mask_token": "<mask>",
   "model_max_length": 512,
   "pad_token": "<pad>",

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:277dadcd95863795c76b3935bf610ec3e8a6f38556616fd5456bf0636a6f89f8
-size 5240
+oid sha256:c1a79038003e0dc41bd6dd3caf4fa825440b4c52727dae834b0b6ada4836c2c6
+size 5368