immisoBETO
immisoBETO is a domain adaptation of a Spanish BERT language model, specifically adapted to the immigrant and misogyny domain.
It was adapted using a guided lexical masking strategy during masked language model (MLM) pretraining. Instead of randomly masking tokens, we prioritized masking words appearing in a immigrant and misogyny-specific lexicon. The base corpus used for domain adaptation was the Spanish Hate Speech Superset.
For training the model we used a batch size of 8, with a learning rate of 2e-5. We trained the model for four epochs using a NVIDIA GeForce RTX 5090 GPU.
Usage
from transformers import pipeline
pipe = pipeline("fill-mask", model="citiusLTL/immisoBETO")
text = pipe("Los [MASK] son los causantes del aumento del desempleo")
print(text)
Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("citiusLTL/immisoBETO")
model = AutoModelForMaskedLM.from_pretrained("citiusLTL/immisoBETO")
- Downloads last month
- 28
Model tree for citiusLTL/immisoBETO
Base model
dccuchile/bert-base-spanish-wwm-uncased