gplsi/SocialTOX
Viewer β’ Updated β’ 4.01k β’ 33 β’ 1
How to use gplsi/Toxicity_model with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="gplsi/Toxicity_model") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("gplsi/Toxicity_model")
model = AutoModelForSequenceClassification.from_pretrained("gplsi/Toxicity_model")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:
The model was fine-tuned on the SocialTOX dataset, 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. In this case, a multiclass classifier was developed.
Base model
PlanTL-GOB-ES/roberta-base-bne