dzarabert / README.md
Sifal's picture
Update README.md
091a0de
|
raw
history blame
1.67 kB
metadata
language:
  - dza
tags:
  - pytorch
  - bert
  - ar
  - dz
license: apache-2.0
widget:
  - text: ' أنا من الجزائر من ولاية [MASK] '
  - text: ' ربي [MASK] خويا لعزيز'
inference: true
drawing

Dzarabert

DzarbiBert is a pruned model of first Transformer-based Language Model that has been pre-trained specifically for the Algerian Dialect (DziriBERT). This pruned version handles Algerian text contents written using Arabic letters. It sets new state of the art results on Algerian text classification datasets, even if it has been pre-trained on much less data (~1 million tweets).

For more information, please visit the paper of the base model: https://arxiv.org/pdf/2109.12346.pdf.

How to use

from transformers import BertTokenizer, BertForMaskedLM

tokenizer = BertTokenizer.from_pretrained("Sifal/dzarabert ")
model = BertForMaskedLM.from_pretrained("Sifal/dzarabert ")

Limitations

The pre-training data used in the base model comes from social media (Twitter). Therefore, the Masked Language Modeling objective may predict offensive words in some situations. Modeling this kind of words may be either an advantage (e.g. when training a hate speech model) or a disadvantage (e.g. when generating answers that are directly sent to the end user). Depending on your downstream task, you may need to filter out such words especially when returning automatically generated text to the end user.