aisyahhrazak
commited on
Commit
•
97d4402
1
Parent(s):
a9bed31
Update README.md
Browse files
README.md
CHANGED
@@ -10,4 +10,31 @@ Safe for Work Classifier Model for Malaysian Data
|
|
10 |
Finetuned https://huggingface.co/mesolitica/malaysian-mistral-191M-MLM-512 with Malaysian NSFW data.
|
11 |
|
12 |
|
13 |
-
![Image in a markdown cell](https://github.com/mesolitica/malaysian-llmops/raw/main/e2e.png)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
Finetuned https://huggingface.co/mesolitica/malaysian-mistral-191M-MLM-512 with Malaysian NSFW data.
|
11 |
|
12 |
|
13 |
+
![Image in a markdown cell](https://github.com/mesolitica/malaysian-llmops/raw/main/e2e.png)
|
14 |
+
|
15 |
+
|
16 |
+
|
17 |
+
### How to use
|
18 |
+
|
19 |
+
```python
|
20 |
+
from classifier import MistralForSequenceClassification
|
21 |
+
model = MistralForSequenceClassification.from_pretrained('malaysia-ai/malaysian-sfw-classifier')
|
22 |
+
```
|
23 |
+
|
24 |
+
|
25 |
+
```
|
26 |
+
precision recall f1-score support
|
27 |
+
|
28 |
+
racist 0.89222 0.90653 0.89931 1808
|
29 |
+
religion insult 0.89681 0.88123 0.88895 3570
|
30 |
+
psychiatric or mental illness 0.93524 0.86953 0.90119 5580
|
31 |
+
sexist 0.77826 0.80563 0.79171 1564
|
32 |
+
harassment 0.79693 0.78777 0.79232 2436
|
33 |
+
informative 0.65000 0.76098 0.70112 1640
|
34 |
+
safe for work 0.73716 0.78421 0.75996 1849
|
35 |
+
|
36 |
+
accuracy 0.84100 18447
|
37 |
+
macro avg 0.81237 0.82798 0.81922 18447
|
38 |
+
weighted avg 0.84680 0.84100 0.84304 18447
|
39 |
+
|
40 |
+
```
|