Views
No views yet
BaoNhan/BN-NewsBERT for binary Vietnamese clickbait detection.StratifiedGroupKFold with seed 42.True.| Metric | Mean ± sample std |
|---|---|
| Test Macro-F1 | 0.7866 ± 0.0058 |
| Test accuracy | 0.8109 ± 0.0074 |
| Dev Macro-F1 | 0.8059 ± 0.0092 |
| seed | dev_macro_f1 | test_macro_f1 | test_accuracy |
|---|---|---|---|
| 22 | 0.8027 | 0.7926 | 0.8099 |
| 42 | 0.7986 | 0.781 | 0.8187 |
| 202 | 0.8162 | 0.7862 | 0.8041 |
0: non-clickbait1: clickbait1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3model_id = "BaoNhan/bn-newsbert-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])