Update README.md
Browse files# 🧠 Toxicity_model_RoBERTa-base-bne– Spanish Toxicity Classifier (Fine-tuned)
## 📌 Model Description
This model is a fine-tuned version** of `RoBERTa-base-bne`, specifically trained to classify the toxicity level of **Spanish-language user comments on news articles**. It distinguishes between three categories:
- **Non-toxic**
-**Slightly-toxic**
- **Toxic**
## 📂 Training Data
The model was fine-tuned on the **[SocialTOX dataset](https://huggingface.co/datasets/gplsi/SocialTOX)**, a collection of Spanish-language comments annotated for varying levels of toxicity. These comments come from news platforms and represent real-world scenarios of online discourse.
---
## Training hyperparameters
- epochs: 7
- learning_rate: 1.51e-6
- beta1: 0.9
- beta2: 0.95
-Adam_epsilon: 2.8e-8
- weight_decay: 3.88e-12
- batch_size: 16
- max_seq_length: 512
@@ -5,6 +5,8 @@ language:
|
|
5 |
metrics:
|
6 |
- f1
|
7 |
- accuracy
|
|
|
|
|
8 |
base_model:
|
9 |
- PlanTL-GOB-ES/roberta-base-bne
|
10 |
pipeline_tag: text-classification
|
|
|
5 |
metrics:
|
6 |
- f1
|
7 |
- accuracy
|
8 |
+
- precision
|
9 |
+
- recall
|
10 |
base_model:
|
11 |
- PlanTL-GOB-ES/roberta-base-bne
|
12 |
pipeline_tag: text-classification
|