Hugging Face's logo

malaysia-ai
/
malaysian-sfw-classifier

Text Classification
Transformers
Safetensors
Malay
mistral
custom_code
text-generation-inference
Inference Endpoints
Model card Files Community
1
malaysia-ai2020 commited on
Commit
4196e78
·
verified ·
1 Parent(s): f465a07

Upload MistralForSequenceClassification

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. config.json +11 -8
  3. model.safetensors +2 -2
README.md CHANGED
@@ -1,7 +1,7 @@
1
  ---
2
- library_name: transformers
3
  language:
4
  - ms
 
5
  ---
6
 
7
  Safe for Work Classifier Model for Malaysian Data(WORK IN PROGRESS)
 
1
  ---
 
2
  language:
3
  - ms
4
+ library_name: transformers
5
  ---
6
 
7
  Safe for Work Classifier Model for Malaysian Data(WORK IN PROGRESS)
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "mesolitica/malaysian-mistral-191M-MLM-512",
3
  "architectures": [
4
  "MistralForSequenceClassification"
5
  ],
@@ -18,18 +18,20 @@
18
  "3": "sexist",
19
  "4": "harassment",
20
  "5": "informative",
21
- "6": "safe for work"
 
22
  },
23
  "initializer_range": 0.02,
24
  "intermediate_size": 3072,
25
  "label2id": {
26
- "harassment": 3,
27
- "informative": 4,
28
- "psychiatric or mental illness": 1,
 
29
  "racist": 0,
30
  "religion insult": 1,
31
- "safe for work": 5,
32
- "sexist": 2
33
  },
34
  "max_position_embeddings": 4096,
35
  "model_type": "mistral",
@@ -52,7 +54,8 @@
52
  "sexist",
53
  "harassment",
54
  "informative",
55
- "safe for work"
 
56
  ],
57
  "vocab_size": 32000
58
  }
 
1
  {
2
+ "_name_or_path": "sfw-mistral-mlm",
3
  "architectures": [
4
  "MistralForSequenceClassification"
5
  ],
 
18
  "3": "sexist",
19
  "4": "harassment",
20
  "5": "informative",
21
+ "6": "safe for work",
22
+ "7": "hate"
23
  },
24
  "initializer_range": 0.02,
25
  "intermediate_size": 3072,
26
  "label2id": {
27
+ "harassment": 4,
28
+ "hate": 7,
29
+ "informative": 5,
30
+ "psychiatric or mental illness": 2,
31
  "racist": 0,
32
  "religion insult": 1,
33
+ "safe for work": 6,
34
+ "sexist": 3
35
  },
36
  "max_position_embeddings": 4096,
37
  "model_type": "mistral",
 
54
  "sexist",
55
  "harassment",
56
  "informative",
57
+ "safe for work",
58
+ "hate"
59
  ],
60
  "vocab_size": 32000
61
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a620dbb8ab91659713db602297762a4f33ed93ef90ad23ab5cfb929bcb24c31c
3
- size 664674264
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c00de7ca5b461ca6f6c185f7945bf86127dbc5ca8804f6050f978093bab6c317
3
+ size 664677336