Views
No views yet
FPTAI/vibert-base-cased for binary Vietnamese clickbait detection.StratifiedGroupKFold with seed 42.True.| Metric | Mean ± sample std |
|---|---|
| Test Macro-F1 | 0.7044 ± 0.0104 |
| Test accuracy | 0.7193 ± 0.0117 |
| Dev Macro-F1 | 0.7348 ± 0.0040 |
| seed | dev_macro_f1 | test_macro_f1 | test_accuracy |
|---|---|---|---|
| 22 | 0.7392 | 0.7137 | 0.731 |
| 42 | 0.7335 | 0.7063 | 0.7193 |
| 202 | 0.7315 | 0.6932 | 0.7076 |
0: non-clickbait1: clickbait1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3model_id = "BaoNhan/vibert-base-cased-ViClickbait-2025"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForSequenceClassification.from_pretrained(model_id)
6title = "Tiêu đề bài báo"
7lead = "Đoạn dẫn của bài báo"
8inputs = tokenizer(title, lead, return_tensors="pt", truncation=True, max_length=256)
9prediction = model(**inputs).logits.argmax(dim=-1).item()
10print(model.config.id2label[prediction])