Instructions to use HiTZ/JaunBERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use HiTZ/JaunBERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="HiTZ/JaunBERT")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("HiTZ/JaunBERT") model = AutoModelForMaskedLM.from_pretrained("HiTZ/JaunBERT", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 2,531 Bytes
8c6ea77 86f86fa 94e9f46 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 94e9f46 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa 8c6ea77 86f86fa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | ---
library_name: transformers
license: apache-2.0
base_model:
- BSC-LT/MrBERT
---
# JaunBERT
## Model Description
JaunBERT is a family of Basque encoder-only language models.
- **Developed by:** HiTZ Research Center & IXA Research group (University of the Basque Country UPV/EHU)
- **Funded by:** IKERGAITU and ALIA projects (Basque and Spanish Governments)
- **License:** Apache 2.0
- **Model Type**: Encoder-only Transformer models (ModernBERT-style)
- **Languages**: Basque (Euskara)
## Getting Started
You can either use this model directly as the example below, or fine-tune it to your task of interest.
```python
>>> from transformers import pipeline
>>> pipe = pipeline("fill-mask", model='HiTZ/JaunBERT')
>>> pipe("Kaixo! Ni <mask> naiz!")
[{'score': 0.034423828125,
'token': 71333,
'token_str': 'Aitor',
'sequence': 'Kaixo! Ni Aitor naiz!'},
{'score': 0.032470703125,
'token': 119830,
'token_str': 'Asier',
'sequence': 'Kaixo! Ni Asier naiz!'},
{'score': 0.0208740234375,
'token': 109222,
'token_str': 'Unai',
'sequence': 'Kaixo! Ni Unai naiz!'},
{'score': 0.0196533203125,
'token': 164657,
'token_str': 'Nerea',
'sequence': 'Kaixo! Ni Nerea naiz!'},
{'score': 0.0196533203125,
'token': 40163,
'token_str': 'Ane',
'sequence': 'Kaixo! Ni Ane naiz!'}]
```
## Training Data
The JaunBERT family was pre-trained on `latxa-corpus-v2`.
Each revision was further trained **on** a specific sub-corpus:
- **`JaunBERT-legala`**: BOPV + BOG + BOTHA + ParlEus
- **`JaunBERT-periodistikoa`**: Euscrawl-v2
- **`JaunBERT-zientifikoa`**: Aldizkariak
Training objective is masked language modeling (MLM) on encoder-only architectures.
## Evaluation
| | **AVG overall** |
|-----------------------|:---------------:|
| JaunBERT | 64.37 |
| JaunBERT-legala | 55.24 |
| JaunBERT-periodistikoa| 66.86 |
| JaunBERT-zientifikoa | 63.53 |
## Acknowledgments
This work has been partially supported by the Basque Government (Research group funding IT1570-22 and IKER-GAITU project), the Spanish Ministry for Digital Transformation and Civil Service, and the EU-funded NextGenerationEU Recovery, Transformation and Resilience Plan (ILENIA project, 2022/TL22/00215335; and ALIA project). Ekhi Azurmendi holds a PhD grant from the Basque Government (PRE_2024_1_0035, respectively). The models were trained on the Leonardo supercomputer at CINECA under the EuroHPC Joint Undertaking, project EHPC-EXT-2024E01-042. |