indobenchmark
/
indobert-base-p1
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
indobenchmark/indobert-base-p1 is a forked repo from huggingface. License: mit
10 Commits 1 Branch 0 Tags 1.5 GiB
Text 100%
Go to file
Patrick von Platen c2cd0b51dd upload flax model 2021-05-19 20:22:23 +00:00
.gitattributes allow flax 2021-05-19 20:22:05 +00:00
README.md Migrate model card from transformers-repo 2020-12-11 22:45:37 +01:00
config.json Update config.json 2020-09-14 19:06:35 +00:00
flax_model.msgpack upload flax model 2021-05-19 20:22:23 +00:00
pytorch_model.bin Update pytorch_model.bin 2020-09-14 19:06:46 +00:00
special_tokens_map.json Update special_tokens_map.json 2020-09-14 19:06:41 +00:00
tf_model.h5 Update tf_model.h5 2020-09-29 16:45:46 +00:00
tokenizer_config.json Update tokenizer_config.json 2020-09-14 19:06:40 +00:00
vocab.txt Update vocab.txt 2020-09-14 19:06:38 +00:00

README.md

language tags license inference datasets
id
indobert
indobenchmark
indonlu
mit false
Indo4B

IndoBERT Base Model (phase1 - uncased)

IndoBERT is a state-of-the-art language model for Indonesian based on the BERT model. The pretrained model is trained using a masked language modeling (MLM) objective and next sentence prediction (NSP) objective.

All Pre-trained Models

Model #params Arch. Training data
indobenchmark/indobert-base-p1 124.5M Base Indo4B (23.43 GB of text)
indobenchmark/indobert-base-p2 124.5M Base Indo4B (23.43 GB of text)
indobenchmark/indobert-large-p1 335.2M Large Indo4B (23.43 GB of text)
indobenchmark/indobert-large-p2 335.2M Large Indo4B (23.43 GB of text)
indobenchmark/indobert-lite-base-p1 11.7M Base Indo4B (23.43 GB of text)
indobenchmark/indobert-lite-base-p2 11.7M Base Indo4B (23.43 GB of text)
indobenchmark/indobert-lite-large-p1 17.7M Large Indo4B (23.43 GB of text)
indobenchmark/indobert-lite-large-p2 17.7M Large Indo4B (23.43 GB of text)

How to use

Load model and tokenizer

from transformers import BertTokenizer, AutoModel
tokenizer = BertTokenizer.from_pretrained("indobenchmark/indobert-base-p1")
model = AutoModel.from_pretrained("indobenchmark/indobert-base-p1")

Extract contextual representation

x = torch.LongTensor(tokenizer.encode('aku adalah anak [MASK]')).view(1,-1)
print(x, model(x)[0].sum())

Authors

IndoBERT was trained and evaluated by Bryan Wilie*, Karissa Vincentio*, Genta Indra Winata*, Samuel Cahyawijaya*, Xiaohong Li, Zhi Yuan Lim, Sidik Soleman, Rahmad Mahendra, Pascale Fung, Syafri Bahar, Ayu Purwarianti.

Citation

If you use our work, please cite:

@inproceedings{wilie2020indonlu,
  title={IndoNLU: Benchmark and Resources for Evaluating Indonesian Natural Language Understanding},
  author={Bryan Wilie and Karissa Vincentio and Genta Indra Winata and Samuel Cahyawijaya and X. Li and Zhi Yuan Lim and S. Soleman and R. Mahendra and Pascale Fung and Syafri Bahar and A. Purwarianti},
  booktitle={Proceedings of the 1st Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics and the 10th International Joint Conference on Natural Language Processing},
  year={2020}
}