rsepulvedat commited on
Commit
6458120
·
verified ·
1 Parent(s): f510b1c

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

Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -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