Hugging Face's logo

rsepulvedat
/
Toxicity_model

Text Classification
Transformers
PyTorch
Spanish
roberta
Inference Endpoints
Model card Files Community
rsepulvedat commited on
Commit
092644b
•
1 Parent(s): eb96b79

Upload config.json

Browse files

update config.json file

Files changed (1) hide show
  1. config.json +10 -9
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_name_or_path": "PlanTL-GOB-ES/roberta-base-bne",
3
  "architectures": [
4
- "RobertaForSequenceClassificationFeatures"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
  "bos_token_id": 0,
@@ -12,19 +12,20 @@
12
  "hidden_dropout_prob": 0.0,
13
  "hidden_size": 768,
14
  "id2label": {
15
- "0": "LABEL_0",
16
- "1": "LABEL_1",
17
- "2": "LABEL_2",
18
- "3": "LABEL_3"
19
  },
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
- "LABEL_0": 0,
24
- "LABEL_1": 1,
25
- "LABEL_2": 2,
26
- "LABEL_3": 3
27
  },
 
28
  "layer_norm_eps": 1e-05,
29
  "max_position_embeddings": 514,
30
  "model_type": "roberta",
 
1
  {
2
  "_name_or_path": "PlanTL-GOB-ES/roberta-base-bne",
3
  "architectures": [
4
+ "RobertaForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
  "bos_token_id": 0,
 
12
  "hidden_dropout_prob": 0.0,
13
  "hidden_size": 768,
14
  "id2label": {
15
+ "0": "No tóxico",
16
+ "1": "Ligeramente tóxico",
17
+ "2": "Tóxico",
18
+ "3": "Muy tóxico"
19
  },
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
+ "No tóxico": 0,
24
+ "Ligeramente tóxico": 1,
25
+ "Tóxico": 2,
26
+ "Muy tóxico": 3
27
  },
28
+
29
  "layer_norm_eps": 1e-05,
30
  "max_position_embeddings": 514,
31
  "model_type": "roberta",