Views
No views yet
xlm-roberta-base fine-tuned for UIT-ViON online-news topic classification on UIT-ViON.HEALTH, correcting the HEATH typo in the repository README without changing the class ID.| Metric | Mean ± std |
|---|---|
| Test Macro-F1 | 0.8043 ± 0.0006 |
| Test accuracy | 0.8059 ± 0.0005 |
| Test macro precision | 0.8044 ± 0.0004 |
| Test macro recall | 0.8059 ± 0.0005 |
| Development Macro-F1 | 0.8015 ± 0.0012 |
| seed | dev_macro_f1 | test_macro_f1 | test_accuracy | micro_batch_size | gradient_accumulation_steps |
|---|---|---|---|---|---|
| 22.000000 | 0.800327 | 0.803955 | 0.805731 | 8.000000 | 1.000000 |
| 42.000000 | 0.802725 | 0.803903 | 0.805538 | 8.000000 | 1.000000 |
| 202.000000 | 0.801385 | 0.804895 | 0.806423 | 8.000000 | 1.000000 |
1{
2 "0": "TECHNOLOGY",
3 "1": "TRAVEL",
4 "2": "EDUCATION",
5 "3": "ENTERTAINMENT",
6 "4": "SCIENCE",
7 "5": "BUSINESS",
8 "6": "LAW",
9 "7": "HEALTH",
10 "8": "WORLD",
11 "9": "SPORT",
12 "10": "NEWS",
13 "11": "VEHICLE",
14 "12": "LIFE"
15}1import torch
2from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
4model_id = "BaoNhan/xlm-roberta-base-UIT-ViON"
5tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
6model = AutoModelForSequenceClassification.from_pretrained(model_id)
7
8text = "Đây là nội dung tiếng Việt cần phân loại."
9inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=64)
10with torch.no_grad():
11 probabilities = model(**inputs).logits.softmax(dim=-1)[0]
12predicted_id = int(probabilities.argmax())
13print(model.config.id2label[predicted_id], probabilities.tolist())aggregate_metrics.json: aggregate metrics and training manifest.artifacts/per_seed_results.csv: one row per fine-tuning seed.artifacts/seed_*_confusion_matrix.csv: confusion matrix for each seed.artifacts/seed_*_classification_report.json: per-class metrics.artifacts/seed_*_test_predictions.csv: IDs, gold/predicted labels and probabilities; raw text is excluded.1@inproceedings{tran2021uitvion,
2 author={Tran, Khanh Quoc and Trinh, Phap Ngoc and Tran, Khoa Nguyen-Anh and Le, An Tran-Hoai and Ha, Luan Van and Nguyen, Kiet Van},
3 title={An Empirical Investigation of Online News Classification on an Open-domain, Large-scale and High-quality Dataset in Vietnamese},
4 booktitle={New Trends in Intelligent Software Methodologies, Tools and Techniques},
5 year={2021},
6 pages={367--379},
7 publisher={IOS Press},
8 doi={10.3233/FAIA210036}
9}