manhteky123 commited on
Commit
2d50b3a
·
verified ·
1 Parent(s): 02062aa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -72
README.md CHANGED
@@ -1,72 +0,0 @@
1
- ---
2
- language: vi
3
- tags:
4
- - vietnamese
5
- - text-classification
6
- - sentiment-analysis
7
- - PhoBERT
8
- - transformers
9
- license: mit
10
- datasets:
11
- - vanhai123/vietnamese-social-comments
12
- metrics:
13
- - accuracy
14
- - f1
15
- model-index:
16
- - name: PhoBERT Vietnamese Comment Classifier (4-class)
17
- results:
18
- - task:
19
- type: text-classification
20
- name: Text Classification
21
- dataset:
22
- name: Vietnamese Social Comments
23
- type: vanhai123/vietnamese-social-comments
24
- metrics:
25
- - type: accuracy
26
- value: 0.86
27
- - type: f1
28
- name: f1_macro
29
- value: 0.83
30
- ---
31
-
32
-
33
- # 📄 PhoBERT Vietnamese Comment Classifier (4-class)
34
-
35
- Đây là mô hình phân loại bình luận tiếng Việt thành 4 nhãn cảm xúc sử dụng `vinai/phobert-base`.
36
-
37
- ## 🍿️ Các nhãn phân loại
38
-
39
- * `positive` – tích cực
40
- * `negative` – tiêu cực
41
- * `neutral` – trung lập
42
- * `toxic` – kích động, phản cảm
43
-
44
- ## 🧠 Mô hình nền
45
-
46
- * **Base model**: [`vinai/phobert-base`](https://huggingface.co/vinai/phobert-base)
47
- * **Fine-tuned** trên dataset `vanhai123/vietnamese-social-comments` gồm 4,896 bình luận từ TikTok, Facebook, YouTube.
48
-
49
- ## 🧪 Kết quả đánh giá
50
-
51
- * Accuracy: **86%**
52
- * Macro F1-score: **83%**
53
-
54
- ## 💻 Sử dụng
55
-
56
- ```python
57
- from transformers import pipeline
58
-
59
- classifier = pipeline("text-classification", model="vanhai123/phobert-vi-comment-4class")
60
- classifier("Video này thật sự rất bổ ích và thú vị!")
61
- ```
62
-
63
- ## 📾 Dataset
64
-
65
- * [Vietnamese Social Comments dataset](https://huggingface.co/datasets/vanhai123/vietnamese-social-comments)
66
-
67
- ## 👤 Tác giả
68
-
69
- * Hà Văn Hải – [vanhai11203@gmail.com](mailto:vanhai11203@gmail.com)
70
- * Hugging Face: [vanhai123](https://huggingface.co/vanhai123)
71
-
72
- ##