uitnlp/vietnamese_students_feedback
Viewer • Updated • 16.2k • 494 • 31
How to use BaoNhan/mbert-UITVSFC-S with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="BaoNhan/mbert-UITVSFC-S") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("BaoNhan/mbert-UITVSFC-S")
model = AutoModelForSequenceClassification.from_pretrained("BaoNhan/mbert-UITVSFC-S", device_map="auto")This model is bert-base-multilingual-cased fine-tuned for UITVSFC-S on UIT-VSFC.
Test metrics are reported as mean ± sample standard deviation over seeds [42, 22, 202].
| Metric | Mean ± std |
|---|---|
| Test Macro-F1 | 0.8019 ± 0.0089 |
| Test accuracy | 0.9388 ± 0.0019 |
| Test macro precision | 0.8365 ± 0.0198 |
| Test macro recall | 0.7813 ± 0.0155 |
| Development Macro-F1 | 0.8137 ± 0.0031 |
| seed | dev_macro_f1 | test_macro_f1 | test_accuracy | micro_batch_size | gradient_accumulation_steps |
|---|---|---|---|---|---|
| 22.000000 | 0.811252 | 0.799441 | 0.940667 | 8.000000 | 1.000000 |
| 42.000000 | 0.812765 | 0.794455 | 0.936959 | 8.000000 | 1.000000 |
| 202.000000 | 0.817189 | 0.811671 | 0.938813 | 8.000000 | 1.000000 |
{
"0": "negative",
"1": "neutral",
"2": "positive"
}
import torch
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model_id = "BaoNhan/mbert-UITVSFC-S"
tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
model = AutoModelForSequenceClassification.from_pretrained(model_id)
text = "Đây là nội dung tiếng Việt cần phân loại."
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=256)
with torch.no_grad():
probabilities = model(**inputs).logits.softmax(dim=-1)[0]
predicted_id = int(probabilities.argmax())
print(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.UIT-VSFC contains student feedback from a specific educational context. Performance may not transfer to other institutions, domains, informal writing styles, or newly emerging vocabulary. Predictions should not be treated as explanations of student intent.
@InProceedings{8573337,
author={Nguyen, Kiet Van and Nguyen, Vu Duc and Nguyen, Phu X. V. and Truong, Tham T. H. and Nguyen, Ngan Luu-Thuy},
booktitle={2018 10th International Conference on Knowledge and Systems Engineering (KSE)},
title={UIT-VSFC: Vietnamese Students' Feedback Corpus for Sentiment Analysis},
year={2018},
pages={19--24},
doi={10.1109/KSE.2018.8573337}
}
Base model
google-bert/bert-base-multilingual-cased