Update labels

README.md

@@ -1,13 +1,11 @@
 ---
-license: apache-2.0
+language: es
 tags:
 - generated_from_trainer
-datasets:
-- null
 metrics:
 - accuracy
 model-index:
-- name: roberta-base-bne-finetuned-mnli
+- name: roberta-base-bne-finetuned-ciberbullying-spanish
   results:
   - task:
       name: Text Classification
@@ -18,27 +16,18 @@ model-index:
       value: 0.9607097303206997
 ---
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
+# roberta-base-bne-finetuned-ciberbullying-spanish
 
-# roberta-base-bne-finetuned-mnli
+This model is a fine-tuned version of [BSC-TeMU/roberta-base-bne](https://huggingface.co/BSC-TeMU/roberta-base-bne) on the dataset generated scrapping all social networks (Twitter, Youtube ...) to detect ciberbullying on Spanish.
 
-This model is a fine-tuned version of [BSC-TeMU/roberta-base-bne](https://huggingface.co/BSC-TeMU/roberta-base-bne) on the None dataset.
 It achieves the following results on the evaluation set:
+
 - Loss: 0.1657
 - Accuracy: 0.9607
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
 ## Training and evaluation data
 
-More information needed
+We use the concatenation from multiple datasets generated scrapping social networks (Twitter,Youtube,Discord...)  to fine-tune this model. The total number of sentence pairs is above 360k sentences.
 
 ## Training procedure
 
@@ -55,6 +44,8 @@ The following hyperparameters were used during training:
 
 ### Training results
 
+<details>
+
 | Training Loss | Epoch | Step  | Accuracy | Validation Loss |
 |:-------------:|:-----:|:-----:|:--------:|:---------------:|
 | 0.1512        | 1.0   | 22227 | 0.9501   | 0.1418          |
@@ -62,6 +53,7 @@ The following hyperparameters were used during training:
 | 0.0973        | 3.0   | 66681 | 0.9594   | 0.1397          |
 | 0.0658        | 4.0   | 88908 | 0.9607   | 0.1657          |
 
+</details>
 
 ### Framework versions
 

config.json

@@ -3,6 +3,14 @@
   "architectures": [
     "RobertaForSequenceClassification"
   ],
+  "id2label": {
+    "0": "Not_bullying",
+    "1": "Bullying"
+  },
+  "label2id": {
+    "Not_bullying": 0,
+    "Bullying": 1
+  },
   "attention_probs_dropout_prob": 0.0,
   "bos_token_id": 0,
   "classifier_dropout": null,