transformers/model_cards/huseinzol05/bert-base-bahasa-cased

language

language
malay

Bahasa BERT Model

Pretrained BERT base language model for Malay and Indonesian.

Pretraining Corpus

bert-base-bahasa-cased model was pretrained on ~1.8 Billion words. We trained on both standard and social media language structures, and below is list of data we trained on,

1. dumping wikipedia.
2. local instagram.
3. local twitter.
4. local news.
5. local parliament text.
6. local singlish/manglish text.
7. IIUM Confession.
8. Wattpad.
9. Academia PDF.

Preprocessing steps can reproduce from here, Malaya/pretrained-model/preprocess.

Pretraining details

• This model was trained using Google BERT's github repository on 3 Titan V100 32GB VRAM.
• All steps can reproduce from here, Malaya/pretrained-model/bert.

Load Pretrained Model

You can use this model by installing torch or tensorflow and Huggingface library transformers. And you can use it directly by initializing it like this:

from transformers import XLNetTokenizer, BertModel

model = BertModel.from_pretrained('huseinzol05/bert-base-bahasa-cased')
tokenizer = XLNetTokenizer.from_pretrained('huseinzol05/bert-base-bahasa-cased')

We use google/sentencepiece to train the tokenizer, so to use it, need to load from XLNetTokenizer.

Results

For further details on the model performance, simply checkout accuracy page from Malaya, https://malaya.readthedocs.io/en/latest/Accuracy.html, we compared with traditional models.

Acknowledgement

Thanks to Im Big, LigBlou, Mesolitica and KeyReply for sponsoring AWS, Google and GPU clouds to train BERT for Bahasa.