talhaa/harmful-content-classifier

1{
2  "add_cross_attention": false,
3  "architectures": [
4    "RobertaForSequenceClassification"
5  ],
6  "attention_probs_dropout_prob": 0.1,
7  "bos_token_id": 0,
8  "classifier_dropout": null,
9  "dtype": "float32",
10  "eos_token_id": 2,
11  "hidden_act": "gelu",
12  "hidden_dropout_prob": 0.1,
13  "hidden_size": 768,
14  "id2label": {
15    "0": "safe",
16    "1": "offensive",
17    "2": "hate_speech",
18    "3": "violent_threat",
19    "4": "identity_attack"
20  },
21  "initializer_range": 0.02,
22  "intermediate_size": 3072,
23  "is_decoder": false,
24  "label2id": {
25    "hate_speech": 2,
26    "identity_attack": 4,
27    "offensive": 1,
28    "safe": 0,
29    "violent_threat": 3
30  },
31  "layer_norm_eps": 1e-05,
32  "max_position_embeddings": 514,
33  "model_type": "roberta",
34  "num_attention_heads": 12,
35  "num_hidden_layers": 12,
36  "pad_token_id": 1,
37  "problem_type": "single_label_classification",
38  "tie_word_embeddings": true,
39  "transformers_version": "5.8.0",
40  "type_vocab_size": 1,
41  "use_cache": false,
42  "vocab_size": 50265
43}
44