Views
No views yet
distilbert-base-multilingual-cased fine-tuned for UIT-VSMEC emotion recognition on UIT-VSMEC.best_model from seed 22, selected by development Macro-F1.| Metric | Mean ± std |
|---|---|
| Test Macro-F1 | 0.3027 ± 0.0102 |
| Test accuracy | 0.4853 ± 0.0106 |
| Test macro precision | 0.3081 ± 0.0559 |
| Test macro recall | 0.3311 ± 0.0081 |
| Development Macro-F1 | 0.2832 ± 0.0051 |
| seed | dev_macro_f1 | test_macro_f1 | test_accuracy |
|---|---|---|---|
| 22.000000 | 0.288982 | 0.312447 | 0.489177 |
| 42.000000 | 0.281637 | 0.292191 | 0.473304 |
| 202.000000 | 0.279091 | 0.303494 | 0.493506 |
1{
2 "0": "Anger",
3 "1": "Disgust",
4 "2": "Enjoyment",
5 "3": "Fear",
6 "4": "Other",
7 "5": "Sadness",
8 "6": "Surprise"
9}1import torch
2from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
4model_id = "BaoNhan/distilbert-multilingual-UITVSMEC"
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=256)
10with torch.no_grad():
11 probabilities = model(**inputs).logits.softmax(dim=-1)[0]
12
13predicted_id = int(probabilities.argmax())
14print(model.config.id2label[predicted_id], probabilities.tolist())aggregate_metrics.json: aggregate metrics and publishing metadata.artifacts/per_seed_results.csv: available completed-seed results.1@inproceedings{ho-etal-2019-emotion,
2 title={Emotion Recognition for Vietnamese Social Media Text},
3 author={Ho, Vong Anh and Nguyen, Duong Huynh-Cong and Nguyen, Danh Hoang and Pham, Linh Thi-Van and Nguyen, Duc-Vu and Nguyen, Kiet Van and Nguyen, Ngan Luu-Thuy},
4 booktitle={Proceedings of the 16th International Conference of the Pacific Association for Computational Linguistics (PACLING 2019)},
5 year={2019},
6 pages={319--333},
7 url={https://arxiv.org/abs/1911.09339}
8}