Hugging Face's logo

xqewec
/
title_classifier

Text Classification
Transformers
PyTorch
roberta
Inference Endpoints
Model card Files Community
1
xqewec commited on
Commit
22c1e0e
•
1 Parent(s): be12970

label update

Browse files
Files changed (1) hide show
  1. config.json +16 -16
config.json CHANGED
@@ -11,28 +11,28 @@
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
13
  "id2label": {
14
- "0": "LABEL_0",
15
  "1": "cs.CL",
16
- "2": "LABEL_2",
17
- "3": "LABEL_3",
18
- "4": "LABEL_4",
19
- "5": "LABEL_5",
20
- "6": "LABEL_6",
21
- "7": "LABEL_7",
22
- "8": "LABEL_8"
23
  },
24
  "initializer_range": 0.02,
25
  "intermediate_size": 3072,
26
  "label2id": {
27
- "LABEL_0": 0,
28
  "cs.CL": 1,
29
- "LABEL_2": 2,
30
- "LABEL_3": 3,
31
- "LABEL_4": 4,
32
- "LABEL_5": 5,
33
- "LABEL_6": 6,
34
- "LABEL_7": 7,
35
- "LABEL_8": 8
36
  },
37
  "layer_norm_eps": 1e-05,
38
  "max_position_embeddings": 514,
 
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
13
  "id2label": {
14
+ "0": "cs.AI",
15
  "1": "cs.CL",
16
+ "2": "cs.CR",
17
+ "3": "cs.CV",
18
+ "4": "cs.CY",
19
+ "5": "cs.DB",
20
+ "6": "cs.NI",
21
+ "7": "cs.SE",
22
+ "8": "other"
23
  },
24
  "initializer_range": 0.02,
25
  "intermediate_size": 3072,
26
  "label2id": {
27
+ "cs.AI": 0,
28
  "cs.CL": 1,
29
+ "cs.CR": 2,
30
+ "cs.CV": 3,
31
+ "cs.CY": 4,
32
+ "cs.DB": 5,
33
+ "cs.NI": 6,
34
+ "cs.SE": 7,
35
+ "other": 8
36
  },
37
  "layer_norm_eps": 1e-05,
38
  "max_position_embeddings": 514,