aisyahhrazak commited on
Commit
97d4402
1 Parent(s): a9bed31

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -1
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
+ ```