Downloads · 30 days
6
18% of all-time downloads
uvegesistvan/huBERTPlain_v2
huBERTPlain_v2 is a text classification model from uvegesistvan. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as cc-by-nc-4.0.
Cased fine-tuned BERT model for Hungarian, trained on a dataset provided by National Tax and Customs Administration - Hungary (NAV): Public Accessibilty Programme. Refined version of the huBERTPlain ('uvegesistvan/huB…
Downloads · 30 days
6
18% of all-time downloads
All-time downloads
33
Public
Parameters
111M
1.3 GB on disk
Likes
0
Public
Click a slice to open those files.
.pt885 MB · 50%
From the Hugging Face model README
Cased fine-tuned BERT model for Hungarian, trained on a dataset provided by National Tax and Customs Administration - Hungary (NAV): Public Accessibilty Programme. Refined version of the huBERTPlain ('uvegesistvan/huBERTPlain') model. Trainig data cleaned further:
The model can be used as any other (cased) BERT model. It has been tested recognizing "accessible" and "original" sentences, where:
Fine-tuned version of the original huBERT model (SZTAKI-HLT/hubert-base-cc), trained on information materials provided by NAV linguistic experts.
| Class | Precision | Recall | F-Score |
|---|---|---|---|
| Accessible / Label_0 | 0.75 | 0.72 | 0.73 |
| Original / Label_1 | 0.74 | 0.77 | 0.75 |
| accuracy | 0.74 | ||
| macro avg | 0.74 | 0.74 | 0.74 |
| weighted avg | 0.74 | 0.74 | 0.74 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("uvegesistvan/huBERTPlain_v2")
model = AutoModelForSequenceClassification.from_pretrained("uvegesistvan/huBERTPlain_v2")
If you use the model, please cite the following dissertation (to be submitted for workshop discussion):
Bibtex:
@PhDThesis{ Uveges:2024,
author = {{"U}veges, Istv{\'a}n},
title = {K{\"o}z{\'e}rthet{\"o} és automatiz{\'a}ci{\'o} - k{\'i}s{\'e}rletek a jog, term{\'e}szetesnyelv-feldolgoz{\'a}s {\'e}s informatika hat{\'a}r{\'a}n.},
year = {2024},
school = {Szegedi Tudom{\'a}nyegyetem}
}