Views
No views yet
uitnlp/CafeBERT fine-tuned for binary Vietnamese fake-news classification on the text-only ViFN benchmark.| Metric | Mean ± std |
|---|---|
| Test Macro-F1 | 0.9076 ± 0.0070 |
| Test accuracy | 0.9078 ± 0.0071 |
| Test macro precision | 0.9117 ± 0.0083 |
| Test macro recall | 0.9081 ± 0.0071 |
| Development Macro-F1 | 0.8856 ± 0.0215 |
| seed | dev_macro_f1 | test_macro_f1 | test_accuracy | micro_batch_size | gradient_accumulation_steps |
|---|---|---|---|---|---|
| 22.000000 | 0.864265 | 0.914683 | 0.914894 | 1.000000 | 8.000000 |
| 42.000000 | 0.885226 | 0.907634 | 0.907801 | 1.000000 | 8.000000 |
| 202.000000 | 0.907335 | 0.900584 | 0.900709 | 1.000000 | 8.000000 |
1{
2 "0": "0",
3 "1": "1"
4}1import torch
2from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
4model_id = "BaoNhan/cafebert-vifn"
5tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
6model = AutoModelForSequenceClassification.from_pretrained(model_id)
7
8text = "Đây là nội dung tin tức tiếng Việt cần phân loại."
9inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=256)
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: complete 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@article{huynh2025vifn,
2 title={Utilizing Transformer Models To Detect Vietnamese Fake News on Social Media Platforms},
3 author={Huynh, Anh-Tuan and Tran, Phuoc},
4 journal={KSII Transactions on Internet and Information Systems},
5 volume={19},
6 number={2},
7 pages={472--487},
8 year={2025},
9 doi={10.3837/TIIS.2025.02.006}
10}